Searched refs:multidim (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | ir_array.h | 65 explicit Index(absl::Span<llvm::Value* const> multidim,
67 : multidim_(multidim.begin(), multidim.end()) {
71 for (const auto* dim : multidim) {
74 index_type_ = multidim[0]->getType();
80 CHECK(absl::c_all_of(multidim, [&](llvm::Value* v) {
98 Index(absl::Span<llvm::Value* const> multidim, const Shape& shape,
104 std::vector<llvm::Value*> multi_index = multidim(); in AddOffsetToDim()
109 const std::vector<llvm::Value*>& multidim() const { return multidim_; } in multidim() function
112 size_t size() const { return multidim().size(); } in size()
114 llvm::Value* operator[](size_t i) const { return multidim()[i]; }
[all …]
|
| D | ir_array.cc | 32 IrArray::Index::Index(absl::Span<llvm::Value* const> multidim, in Index() argument
35 : Index(multidim, shape, index_type) { in Index()
40 void IrArray::Index::Delinearize(std::vector<llvm::Value*>* multidim, in Delinearize() argument
61 (*multidim)[dimension] = b->CreateURem( in Delinearize()
64 (*multidim)[dimension] = quot; in Delinearize()
84 IrArray::Index::Index(absl::Span<llvm::Value* const> multidim, in Index() argument
86 : multidim_(multidim.begin(), multidim.end()), in Index()
92 CHECK_EQ(shape.dimensions_size(), multidim.size()); in Index()
93 for (const auto* dim : multidim) { in Index()
206 Permute(dimension_mapping, multidim()); in SourceIndexOfTranspose()
|
| D | kernel_tiling.cc | 175 std::vector<llvm::Value*> multidim; in GetTileIndexForBlockOrigin() local
176 multidim.reserve(block_sizes_.size()); in GetTileIndexForBlockOrigin()
178 multidim.push_back(b_->CreateMul( in GetTileIndexForBlockOrigin()
183 return IrArray::Index(multidim, block_index[0]->getType()); in GetTileIndexForBlockOrigin()
188 std::vector<llvm::Value*> elem_multi_index = tile_index.multidim(); in GetElementIndexForTileOrigin()
|
| D | fused_ir_emitter.cc | 46 if (generated_value_cache_[hlo].contains(index.multidim())) { in DefaultAction()
48 generated_value_cache_[hlo][index.multidim()]; in DefaultAction()
74 TF_ASSIGN_OR_RETURN(generated_value_cache_[hlo][index.multidim()], in DefaultAction()
77 return generated_value_cache_[hlo][index.multidim()]; in DefaultAction()
|
| D | sort_util.cc | 190 std::vector<llvm::Value*> keys_multi_index = tiled_keys_index.multidim(); in EmitTiledCompareLoop()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | elemental_ir_emitter.cc | 1771 std::vector<llvm::Value*> operand_multi_index = source_index.multidim(); in EmitElementalConcatenate()
1786 std::vector<llvm::Value*> source_multi_index = source_index.multidim(); in EmitElementalConcatenate()
2054 std::vector<llvm::Value*> multi_index = padded_index.multidim(); in EmitElementalPad()
2288 std::vector<llvm::Value*> source_multi_index = target_index.multidim(); in MakeElementGenerator()
2410 IrArray::Index source_index(target_index.multidim(), in MakeElementGenerator()
|
| /external/deqp-deps/glslang/Test/baseResults/ |
| D | hlsl.array.multidim.frag.out | 1 hlsl.array.multidim.frag
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 3141 Permute({0, 2, 1}, output_tile_origin.multidim())); in EmitKernel()
|