Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/dtensor/mlir/expansions/
Dmeta_spmd_expander.cc486 std::vector<std::string> output_layout_specs; in ComputeLayoutForward() local
915 std::vector<std::string> output_layout_specs; in ComputeLayoutForward() local
1094 std::vector<std::string> output_layout_specs; in ComputeLayoutForward() local
Dgather_spmd_expander.cc194 std::vector<std::string> output_layout_specs; in ComputeLayoutForward() local
/external/tensorflow/tensorflow/dtensor/mlir/
Dspmd_expander_common.cc208 std::vector<string> output_layout_specs; in GetBroadcastLayoutForElementWise() local