Searched defs:to_layout (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/parallel/tensor_layout/ |
| D | tensor_redistribution_test.cc | 82 TensorLayout to_layout; in TEST_F() local
135 TensorLayout to_layout; in TEST_F() local
188 TensorLayout to_layout; in TEST_F() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/tensor_layout/ |
| D | tensor_redistribution.cc | 37 auto to_layout = to.LayoutForRedistribution(); in MakeFromToLayout() local
824 TensorLayout to_layout; in InferTensorRedistributionOperatorList() local
940 … TensorRedistribution::InferReshape(const TensorLayout &from_layout, const TensorLayout &to_layout, in InferReshape()
1031 …Redistribution::InferRedistribution(const TensorLayout &from_layout, const TensorLayout &to_layout, in InferRedistribution()
|
| D | tensor_transform.cc | 176 TensorLayout to_layout; in TransformOperators() local
|
| /third_party/vk-gl-cts/external/amber/src/src/vulkan/ |
| D | transfer_image.cc | 309 VkImageLayout to_layout, in ImageBarrier()
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/parallel/ |
| D | _tensor.py | 342 def _transform_tensor_by_layout(from_layout, to_layout, device_list, rank_id): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | reshape_info.cc | 330 bool SpecialPatternInTransformer(const TensorLayout &from_layout, const TensorLayout &to_layout) { in SpecialPatternInTransformer()
357 bool SkipTensorRedistribution(const TensorLayout &from_layout, const TensorLayout &to_layout) { in SkipTensorRedistribution()
|
| /third_party/rust/rust/compiler/rustc_codegen_cranelift/src/ |
| D | base.rs | 579 let to_layout = fx.layout_of(fx.monomorphize(to_ty)); in codegen_stmt() localVariable
613 let to_layout = fx.layout_of(fx.monomorphize(to_ty)); in codegen_stmt() localVariable
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/graph_util/ |
| D | graph_utils.cc | 358 TensorLayout to_layout = tensor_redistribution->layout_transfer().to_in(); in MatchingAccordingToIndex() local
550 TensorLayout to_layout = tensor_redistribution->layout_transfer().to_in(); in WhetherMatchingIsNeededForReshape() local
|