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