Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer.cc308 Tensor paddings_t; in EraseCancellableNodesAroundPad() local
309 if (!GetValueAttrFromConstInputNode(*pad, IsPad, 1, &paddings_t)) continue; in EraseCancellableNodesAroundPad()
354 auto paddings_s = absl::Span<int32>(paddings_t.flat<int32>().data(), in EraseCancellableNodesAroundPad()
355 paddings_t.NumElements()); in EraseCancellableNodesAroundPad()
362 paddings_t.AsProtoTensorContent(permuted_paddings_tensor.mutable_tensor()); in EraseCancellableNodesAroundPad()
/external/tensorflow/tensorflow/core/ops/
Darray_ops_test.cc459 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()
490 Tensor paddings_t(DT_INT64, TensorShape{3, 2}); in TEST() local
491 test::FillValues<int64>(&paddings_t, {1, 10, 2, 20, 3, 30}); in TEST()
492 op.input_tensors[1] = &paddings_t; in TEST()
523 Tensor paddings_t(DT_INT64, TensorShape{3, 2}); in TEST() local
524 test::FillValues<int64>(&paddings_t, {1, 10, 2, 20, 3, 30}); in TEST()
525 op.input_tensors[1] = &paddings_t; in TEST()
Darray_ops.cc65 const Tensor* paddings_t, int64 num_dims) { in PadKnown() argument
68 auto paddings_data = paddings_t->matrix<T>(); in PadKnown()
97 const Tensor* paddings_t = c->input_tensor(1); in PadShapeFn() local
100 if (paddings_t == nullptr) { in PadShapeFn()
110 const int64 num_dims = paddings_t->shape().dim_size(0); in PadShapeFn()
114 if (paddings_t->dtype() == DT_INT32) { in PadShapeFn()
115 return PadKnown<int32>(c, input, paddings_t, num_dims); in PadShapeFn()
117 return PadKnown<int64>(c, input, paddings_t, num_dims); in PadShapeFn()
1963 const Tensor* paddings_t, int64 input_rank) { in MirrorPadKnown() argument
1964 auto paddings_data = paddings_t->matrix<T>(); in MirrorPadKnown()
[all …]
Dnn_ops.cc457 const Tensor* paddings_t = c->input_tensor(paddings_index); in CommonFusedConvCalculations() local
459 if (paddings_t != nullptr) { in CommonFusedConvCalculations()
463 int64 p0 = static_cast<int64>(paddings_t->matrix<int32>()(i, 0)); in CommonFusedConvCalculations()
464 int64 p1 = static_cast<int64>(paddings_t->matrix<int32>()(i, 1)); in CommonFusedConvCalculations()