Searched refs:CreateScalarConst (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | collection_ops_util.cc | 43 Value CreateScalarConst(int32_t value, OpBuilder builder, Location loc) { in CreateScalarConst() function
73 ArrayRef<Value>{index, zeros_tensor, CreateScalarConst(0, builder, loc)}); in GetIndicesForElement()
158 auto zero = CreateScalarConst(0, builder, op->getLoc()); in CreateInitBufferValue()
286 ArrayRef<Value>{buffer, indices, CreateScalarConst(0, builder, loc)}); in GatherElements()
|
| D | collection_ops_util.h | 36 Value CreateScalarConst(int32_t value, OpBuilder builder, Location loc);
|
| D | tensor_array_ops_decomposition.cc | 372 auto result = cutil::CreateScalarConst(buffer_type.getDimSize(0), builder, in HandleTensorArraySizeV3Op()
|
| D | tensor_list_ops_decomposition.cc | 737 auto dim = cutil::CreateScalarConst( in HandleTensorListLengthOp()
|