Home
last modified time | relevance | path

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

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