Searched refs:paddings_t (Results 1 – 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/core/ops/ |
D | array_ops_test.cc | 428 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()
|
D | array_ops.cc | 58 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 …]
|
D | nn_ops.cc | 377 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()
|