Searched defs:paddings_t (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/ops/
H A Dnn_ops.cc352 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 Darray_ops.cc57 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