Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dcollection_ops_util.h67 LogicalResult CreateInitBufferValue(ArrayRef<int64_t> element_shape,
73 LogicalResult CreateInitBufferValue(ArrayRef<int64_t> element_shape,
Dcollection_ops_util.cc135 LogicalResult CreateInitBufferValue(ArrayRef<int64_t> element_shape, in CreateInitBufferValue() function
145 return CreateInitBufferValue(element_shape, max_size_const, op, element_dtype, in CreateInitBufferValue()
149 LogicalResult CreateInitBufferValue(ArrayRef<int64_t> element_shape, in CreateInitBufferValue() function
Dtensor_array_ops_decomposition.cc203 if (failed(cutil::CreateInitBufferValue(*elem_shape, ta.size(), ta, in HandleTensorArrayV3Op()
389 if (failed(cutil::CreateInitBufferValue( in CreateAndInitializeGradVariable()
Dtensor_list_ops_decomposition.cc577 if (failed(cutil::CreateInitBufferValue( in HandleEmptyTensorListOp()
604 if (failed(cutil::CreateInitBufferValue(element_shape, list.num_elements(), in HandleTensorListReserveOp()
Dstack_ops_decomposition.cc391 if (failed(cutil::CreateInitBufferValue( in HandleStackV2Op()