| /external/tensorflow/tensorflow/dtensor/mlir/ |
| D | layout_parsing.cc | 127 std::vector<Layout> layouts; in ExtractRequiredLayoutFromOp() local
219 std::vector<Layout> layouts; in ExtractRequiredLayoutFromOperands() local
229 absl::Span<const absl::optional<Layout>> layouts) { in SetLayoutOnOp()
241 absl::Span<const absl::optional<Layout>> layouts) { in SetLayoutOnOp()
|
| D | layout_propagation_v2.cc | 70 const llvm::DenseMap<mlir::Value, Layout>& layouts) { in AllOpResultsHaveLayouts()
91 llvm::DenseMap<mlir::Value, Layout>& layouts) { in UpdateLayoutForSkippedOps()
140 llvm::DenseMap<mlir::Value, Layout>& layouts) { in CopyLayoutsForSkippedOps()
462 mlir::LogicalResult GetMostShardedLayout(llvm::ArrayRef<Layout> layouts, in GetMostShardedLayout()
1076 llvm::SmallVector<Layout, 4> layouts{if_result_layout.layout(), in InsertDTensorLayoutForIfRegionOp() local
|
| D | propagate_default_layout.cc | 185 const auto layouts = layout_or_status.ValueOrDie(); in runOnOperation() local
|
| /external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| D | dataparallel_spmd_expander.cc | 50 bool AllReplicated(const std::vector<Layout>& layouts) { in AllReplicated()
58 bool AllBatchParallel(const std::vector<Layout>& layouts, in AllBatchParallel()
66 bool SameBatchRank(const std::vector<Layout>& layouts, in SameBatchRank()
79 bool AnyLayoutExist(const llvm::DenseMap<int, Layout>& layouts, in AnyLayoutExist()
95 const llvm::DenseMap<int, Layout>& layouts, in MergeBatchLayouts()
|
| D | replicated_spmd_expander.cc | 33 bool AllReplicated(const std::vector<Layout>& layouts) { in AllReplicated()
|
| D | split_spmd_expander.cc | 40 int64_t split_dim, const llvm::DenseMap<int, Layout>& layouts) { in MergeLayoutsForSplitOutput()
|
| D | einsum_spmd_expander.cc | 154 bool replicate_incompatible_dimensions, const std::vector<Layout>& layouts, in GetLabelToShardingSpec()
274 std::vector<Layout> layouts; in ComputeLayoutForward() local
|
| D | save_restore_spmd_expander.cc | 542 std::vector<Layout> layouts, std::vector<mlir::Type> output_types, in ExpandRestoreV2OpHelper()
794 llvm::SmallVector<Layout> layouts(restore_op->getNumResults()); in GetLayoutsFromAssignVariableOps() local
|
| /external/angle/src/tests/gl_tests/ |
| D | VulkanImageTest.cpp | 238 GLenum layouts[1] = {GL_NONE}; in TEST_P() local
294 GLenum layouts[1] = {GL_NONE}; in TEST_P() local
362 GLenum layouts[1] = {GL_NONE}; in TEST_P() local
429 GLenum layouts[1] = {GL_NONE}; in TEST_P() local
524 GLenum layouts[1] = {GL_NONE}; in TEST_P() local
|
| /external/pytorch/torch/csrc/jit/codegen/onednn/ |
| D | layout_propagation.cpp | 18 std::vector<int64_t> layouts(num_output, STRIDED_LAYOUT); in LayoutPropagation() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | shape_util.cc | 29 std::vector<int>* layouts) { in PopulateInfeedLayoutVector()
167 std::vector<int> layouts; in GetShapeLayoutVector() local
|
| /external/rust/android-crates-io/crates/vulkano/src/render_pass/ |
| D | macros.rs | 79 let mut layouts: Vec<( localVariable
|
| /external/drm_hwcomposer/bufferinfo/ |
| D | BufferInfoMapperMetadata.cpp | 142 std::vector<ui::PlaneLayout> layouts; in GetBoInfo() local
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | dumped_computation_to_text.cc | 58 std::vector<const Shape*> layouts; in RealMain() local
|
| D | dumped_computation_to_operation_list.cc | 82 std::vector<const Shape*> layouts; in RealMain() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | client_test.cc | 42 std::vector<std::vector<int64_t>> layouts = {{0, 1}, {1, 0}}; in XLA_TEST_F() local
|
| /external/mesa3d/src/util/u_gralloc/ |
| D | u_gralloc_imapper5_api.cpp | 77 std::vector<PlaneLayout> &layouts = *layouts_opt; in mapper5_get_buffer_basic_info() local
|
| D | u_gralloc_imapper4_api.cpp | 136 std::vector<PlaneLayout> &layouts = *layouts_opt; in mapper4_get_buffer_basic_info() local
|
| /external/tensorflow/tensorflow/dtensor/cc/ |
| D | dtensor_sharded_prefix_kernel.cc | 41 const Tensor* layouts; in Compute() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | attribute_importer.cc | 162 std::vector<mlir::Attribute> layouts; in ExtractLayoutsFromShapes() local
|
| /external/pytorch/aten/src/ATen/native/vulkan/api/ |
| D | Descriptor.cpp | 176 std::vector<VkDescriptorSetLayout> layouts(descriptors_.size()); in allocate_new_batch() local
|
| /external/mesa3d/src/freedreno/fdl/ |
| D | fd6_view.c | 144 fdl6_view_init(struct fdl6_view *view, const struct fdl_layout **layouts, in fdl6_view_init()
|
| /external/rust/android-crates-io/crates/vulkano/src/descriptor_set/ |
| D | update.rs | 309 let layouts = image_view.image().descriptor_layouts().expect( in to_vulkan_info() localVariable
330 let layouts = image_view.image().descriptor_layouts().expect( in to_vulkan_info() localVariable
|
| /external/executorch/backends/vulkan/runtime/vk_api/ |
| D | Descriptor.cpp | 233 std::vector<VkDescriptorSetLayout> layouts(descriptors_.size()); in allocate_new_batch() local
|
| /external/swiftshader/tests/VulkanWrapper/ |
| D | DrawTester.cpp | 346 std::vector<vk::DescriptorSetLayout> layouts(1, descriptorSetLayout); in createCommandBuffers() local
|