Home
last modified time | relevance | path

Searched defs:tensor_layout (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/dtensor/mlir/expansions/
Dscatter_spmd_expander.cc33 StatusOr<Layout> GetOutputLayout(const absl::optional<Layout>& tensor_layout, in GetOutputLayout()
164 absl::optional<Layout> tensor_layout; in TensorScatterOpComputeLayoutForward() local
/external/angle/third_party/spirv-tools/src/source/opt/
Dtype_manager.cpp458 auto tensor_layout = type->AsTensorLayoutNV(); in GetTypeInstruction() local
/external/deqp-deps/SPIRV-Tools/source/opt/
Dtype_manager.cpp445 auto tensor_layout = type->AsTensorLayoutNV(); in GetTypeInstruction() local
/external/deqp-deps/SPIRV-Tools/source/val/
Dvalidate_memory.cpp2277 const auto tensor_layout = _.FindDef(tensor_layout_id); in ValidateCooperativeMatrixLoadStoreTensorNV() local
/external/angle/third_party/spirv-tools/src/source/val/
Dvalidate_memory.cpp2260 const auto tensor_layout = _.FindDef(tensor_layout_id); in ValidateCooperativeMatrixLoadStoreTensorNV() local