Home
last modified time | relevance | path

Searched refs:TensorShapeAsVector (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/c/eager/
Dc_api_debug.cc30 std::vector<int64_t> TensorShapeAsVector(const tensorflow::TensorHandle& handle, in TensorShapeAsVector() function
64 std::vector<int64_t> dev_dims = TensorShapeAsVector(*handle, &status->status); in TFE_TensorHandleTensorDebugInfo()
/external/tensorflow/tensorflow/dtensor/cc/
Ddtensor_device_util.cc141 std::vector<int64_t> shape(TensorShapeAsVector(tensor, status)); in BroadcastResourceTensor()
293 std::vector<int64_t> shape(TensorShapeAsVector(tensor, status)); in Broadcast()
640 std::vector<int64_t> TensorShapeAsVector(TFE_TensorHandle* tensor, in TensorShapeAsVector() function
Ddtensor_device_util.h593 std::vector<int64_t> TensorShapeAsVector(TFE_TensorHandle* tensor,
Ddtensor_device.cc1927 std::vector<int64_t> tensor_shape(TensorShapeAsVector(input, status)); in Execute()