Searched refs:ExtractSmallTensorValue (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/ |
| H A D | small_constant_optimization.h | 35 absl::optional<NodeDef> ExtractSmallTensorValue(TFE_Context* context,
|
| H A D | small_constant_optimization.cc | 74 absl::optional<NodeDef> ExtractSmallTensorValue(TFE_Context* context, in ExtractSmallTensorValue() function
|
| H A D | dtensor_device_util.cc | 299 ExtractSmallTensorValue(context, tensor, layout, status); in Broadcast()
322 ExtractSmallTensorValue(context, tensor, layout, status); in Broadcast()
|
| H A D | dtensor_device.cc | 1894 ExtractSmallTensorValue(context, input, t->layout(), status); in Execute()
|