Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_space_to_batch_nd_attributes.cc44 const int paddings_index = 2; in Run() local
48 !IsConstantParameterArray(*model, op->inputs[paddings_index])) in Run()
52 const auto& paddings_array = model->GetArray(op->inputs[paddings_index]); in Run()
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_fused_image_transform.cc670 int paddings_index; in Compute() local
673 paddings_index = 2; in Compute()
676 paddings_index = 1; in Compute()
679 const Tensor& paddings = context->input(paddings_index); in Compute()
/external/tensorflow/tensorflow/core/ops/
Dnn_ops.cc350 int paddings_index = 1; in CommonFusedConvCalculations() local
353 paddings_index = 2; in CommonFusedConvCalculations()
371 TF_RETURN_IF_ERROR(c->WithRank(c->input(paddings_index), 2, &paddings)); in CommonFusedConvCalculations()
377 const Tensor* paddings_t = c->input_tensor(paddings_index); in CommonFusedConvCalculations()