Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/kernel_tests/
Dspacetobatch_op_test.py659 crops_const = tensor_util.constant_value(crops)
661 self.assertIsNotNone(crops_const)
663 paddings_const, crops_const)
684 self.assertAllEqual(crops_result, crops_const)
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Dconv_2d_builder.cc247 auto* crops_const = graph_builder_->AddConstNodeWithData( in PopulateSubGraph() local
316 batch_to_space_op->AddInput(TensorID(crops_const->GetID(), 0)); in PopulateSubGraph()
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_common.cc1113 SmallVector<int32_t, 2> crops_const(2 * (crops_dims)); in convertBatchToSpaceNDOp() local
1116 crops_const[idx++] = i.getInt(); in convertBatchToSpaceNDOp()
1120 int crops_lo = crops_const[i * crops_dims + 0]; in convertBatchToSpaceNDOp()
1121 int crops_hi = crops_const[i * crops_dims + 1]; in convertBatchToSpaceNDOp()