Home
last modified time | relevance | path

Searched defs:global_output_shapes (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dspmd_expander.cc82 llvm::SmallVector<llvm::SmallVector<int64_t, 4>, 4> global_output_shapes; in ExpandOpAndSetLayout() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Ddtensor_device_util.cc658 std::vector<PartialTensorShape>* global_output_shapes, in PrepareGraphForMlir()
848 const std::vector<PartialTensorShape>& global_output_shapes) { in IdentifyAllFunctionsToExecute()
H A Ddtensor_device.cc1078 const std::vector<PartialTensorShape>& global_output_shapes, in UpdateOutputLayoutsWithSameShapePolicy()
1328 std::vector<PartialTensorShape> global_output_shapes; in LowerToSPMDFunction() local