Home
last modified time | relevance | path

Searched refs:GetR1Const (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dcollection_ops_util.cc49 Value GetR1Const(ArrayRef<int64_t> r1, OpBuilder builder, Location loc, in GetR1Const() function
67 auto zeros_tensor = GetR1Const(zeros, builder, loc); in GetIndicesForElement()
84 auto size_const = GetR1Const(slice_size, builder, loc); in GetElement()
97 GetR1Const(element_type.getShape(), builder, loc)}); in GetElement()
114 ArrayRef<Value>{element, GetR1Const(slice_shape, builder, loc)}); in SetElement()
132 ArrayRef<Value>{scalar, GetR1Const(size_type.getShape(), builder, loc)}); in ReshapeScalarToSizeType()
167 ArrayRef<Value>{zero, GetR1Const(buffer_shape, builder, op->getLoc())}); in CreateInitBufferValue()
279 ArrayRef<Value>{buffer, GetR1Const(slice_starts, builder, loc), in GatherElements()
280 GetR1Const(result_shape, builder, loc)}); in GatherElements()
303 auto slice_sizes = GetR1Const(per_slice_shape, builder, loc); in ScatterAccumulateElements()
[all …]
Dcollection_ops_util.h39 Value GetR1Const(ArrayRef<int64_t> r1, OpBuilder builder, Location loc,
Dtensor_list_ops_decomposition.cc582 Value size = cutil::GetR1Const({0LL}, builder, list.getLoc()); in HandleEmptyTensorListOp()
628 Value size = cutil::GetR1Const({type.getShape()[0]}, builder, list.getLoc()); in HandleTensorListFromTensorOp()
653 ArrayRef<Value>{size, cutil::GetR1Const({1LL}, builder, push.getLoc())}); in HandleTensorListPushBackOp()
678 ArrayRef<Value>{size, cutil::GetR1Const({1LL}, builder, pop.getLoc())}); in HandleTensorListPopBackOp()
749 cutil::GetR1Const({}, builder, length.getLoc())}); in HandleTensorListLengthOp()
763 auto result = cutil::GetR1Const( in HandleTensorListElementShapeOp()
Dstack_ops_decomposition.cc407 cutil::GetR1Const({0LL}, builder, stack.getLoc()), in HandleStackV2Op()
437 ArrayRef<Value>{index, cutil::GetR1Const({1}, builder, push.getLoc())}); in HandleStackPushV2Op()
457 ArrayRef<Value>{size, cutil::GetR1Const({1}, builder, pop.getLoc())}); in HandleStackPopV2Op()
Dtensor_array_ops_decomposition.cc272 ArrayRef<Value>{elem, cutil::GetR1Const(slice_type.getShape(), builder, in HandleTensorArrayWriteV3Op()
306 cutil::GetR1Const(shape, builder, concat.getLoc())}); in HandleTensorArrayConcatV3Op()
346 cutil::GetR1Const(buffer_shape, builder, in HandleTensorArraySplitV3Op()