Searched refs:PutValueIntoTensor (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | constant_folding.cc | 325 static Status PutValueIntoTensor(const int64_t value, const DataType& type, in PutValueIntoTensor() function
346 TF_RETURN_IF_ERROR(PutValueIntoTensor(shp.dim_size(i), type, i, tensor)); in ConvertShapeToConstant()
354 TF_RETURN_IF_ERROR(PutValueIntoTensor(size, type, 0, tensor)); in ConvertShapeToConstant()
358 TF_RETURN_IF_ERROR(PutValueIntoTensor(shp.dims(), type, 0, tensor)); in ConvertShapeToConstant()
|