Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/tpu/
Dtpu_embedding_output_layout_utils.h29 Status ComputeOutputTensorShapes(
Dtpu_embedding_output_layout_utils.cc23 Status ComputeOutputTensorShapes( in ComputeOutputTensorShapes() function
/external/tensorflow/tensorflow/core/ops/
Dtpu_embedding_ops.cc73 TF_RETURN_IF_ERROR(ComputeOutputTensorShapes(config, &output_shapes)); in __anon3e675d600102()
/external/tensorflow/tensorflow/core/tpu/ops/
Dtpu_embedding_ops.cc404 TF_RETURN_IF_ERROR(ComputeOutputTensorShapes(config, &output_shapes)); in __anoned4aa6880a02()