Home
last modified time | relevance | path

Searched refs:paddings_t (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/ops/
Darray_ops_test.cc428 Tensor paddings_t(DT_INT64, TensorShape{3, 2}); in TEST() local
429 test::FillValues<int64>(&paddings_t, {1, 10, 2, 20, 3, 30}); in TEST()
430 op.input_tensors[1] = &paddings_t; in TEST()
459 Tensor paddings_t(DT_INT64, TensorShape{3, 2}); in TEST() local
460 test::FillValues<int64>(&paddings_t, {1, 10, 2, 20, 3, 30}); in TEST()
461 op.input_tensors[1] = &paddings_t; in TEST()
492 Tensor paddings_t(DT_INT64, TensorShape{3, 2}); in TEST() local
493 test::FillValues<int64>(&paddings_t, {1, 10, 2, 20, 3, 30}); in TEST()
494 op.input_tensors[1] = &paddings_t; in TEST()
Darray_ops.cc58 const Tensor* paddings_t, int64 num_dims) { in PadKnown() argument
61 auto paddings_data = paddings_t->matrix<T>(); in PadKnown()
90 const Tensor* paddings_t = c->input_tensor(1); in PadShapeFn() local
93 if (paddings_t == nullptr) { in PadShapeFn()
103 const int64 num_dims = paddings_t->shape().dim_size(0); in PadShapeFn()
107 if (paddings_t->dtype() == DT_INT32) { in PadShapeFn()
108 return PadKnown<int32>(c, input, paddings_t, num_dims); in PadShapeFn()
110 return PadKnown<int64>(c, input, paddings_t, num_dims); in PadShapeFn()
1808 const Tensor* paddings_t, int64 input_rank) { in MirrorPadKnown() argument
1809 auto paddings_data = paddings_t->matrix<T>(); in MirrorPadKnown()
[all …]
Dnn_ops.cc377 const Tensor* paddings_t = c->input_tensor(paddings_index); in CommonFusedConvCalculations() local
379 if (paddings_t != nullptr) { in CommonFusedConvCalculations()
383 int64 p0 = static_cast<int64>(paddings_t->matrix<int32>()(i, 0)); in CommonFusedConvCalculations()
384 int64 p1 = static_cast<int64>(paddings_t->matrix<int32>()(i, 1)); in CommonFusedConvCalculations()