Searched defs:paddings_t (Results 1 - 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/core/ops/ |
H A D | nn_ops.cc | 352 const Tensor* paddings_t = c->input_tensor(paddings_index); local 354 if (paddings_t != nullptr) { 358 int64 p0 = static_cast<int64>(paddings_t->matrix<int32>()(i, 0)); 359 int64 p1 = static_cast<int64>(paddings_t->matrix<int32>()(i, 1));
|
H A D | array_ops.cc | 57 const Tensor* paddings_t, int64 num_dims) { 58 // paddings_t is known. 60 auto paddings_data = paddings_t->matrix<T>(); 89 const Tensor* paddings_t = c->input_tensor(1); local 91 // paddings_t is unknown 92 if (paddings_t == nullptr) { 102 const int64 num_dims = paddings_t->shape().dim_size(0); 106 if (paddings_t->dtype() == DT_INT32) { 107 return PadKnown<int32>(c, input, paddings_t, num_dims); 109 return PadKnown<int64>(c, input, paddings_t, num_dim 56 PadKnown(InferenceContext* c, ShapeHandle input, const Tensor* paddings_t, int64 num_dims) argument 1662 MirrorPadKnown(InferenceContext* c, ShapeHandle input, const Tensor* paddings_t, int64 input_rank) argument 1931 SpaceToBatchShapeHelper(InferenceContext* c, ShapeHandle input_shape, ShapeHandle block_shape_shape, const Tensor* block_shape_t, ShapeHandle paddings_shape, const Tensor* paddings_t) argument [all...] |
Completed in 102 milliseconds