Home
last modified time | relevance | path

Searched refs:multidim (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dir_array.h79 Index(absl::Span<llvm::Value* const> multidim, const Shape& shape,
85 Index(absl::Span<llvm::Value* const> multidim,
98 const std::vector<llvm::Value*>& multidim() const { return multidim_; } in multidim() function
102 size_t size() const { return multidim().size(); } in size()
104 llvm::Value* operator[](size_t i) const { return multidim()[i]; }
108 const_iterator begin() const { return multidim().begin(); } in begin()
109 const_iterator end() const { return multidim().end(); } in end()
179 Index(absl::Span<llvm::Value* const> multidim, llvm::Value* linear,
182 void Delinearize(std::vector<llvm::Value*>* multidim, llvm::Value* linear,
186 void Delinearize(std::vector<llvm::Value*>* multidim, llvm::Value* linear,
Dir_array.cc37 IrArray::Index::Index(absl::Span<llvm::Value* const> multidim, in Index() argument
40 : Index(multidim, shape, index_type) { in Index()
45 void IrArray::Index::Delinearize(std::vector<llvm::Value*>* multidim, in Delinearize() argument
66 (*multidim)[dimension] = b->CreateURem( in Delinearize()
69 (*multidim)[dimension] = quot; in Delinearize()
75 void IrArray::Index::Delinearize(std::vector<llvm::Value*>* multidim, in Delinearize() argument
92 (*multidim)[dimension] = in Delinearize()
96 (*multidim)[dimension] = quot; in Delinearize()
130 IrArray::Index::Index(absl::Span<llvm::Value* const> multidim, in Index() argument
133 : Index(multidim, ShapeUtil::MakeShape(/*arbitrary*/ PRED, dimensions), in Index()
[all …]
Dfused_ir_emitter.cc47 generated_value_cache_[hlo], index.multidim(), nullptr)) { in DefaultAction()
76 generated_value_cache_[hlo][index.multidim()] = generated_value; in DefaultAction()
Dsort_util.cc191 std::vector<llvm::Value*> keys_multi_index = tiled_keys_index.multidim(); in EmitTiledCompareLoop()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc4127 auto multidim = normalized_shape_index.multidim(); in GetUnnormalizedIndex() local
4128 return IrArray::Index({multidim[1], multidim[2]}, unnormalized_shape, in GetUnnormalizedIndex()
4452 IrArray::Index output_index(element_index.multidim(), in EmitEpilogueForReduction()
4641 input_index.multidim(), reduction_operand_shape, input_index.GetType()); in EmitTileElementForReduction()
4723 std::vector<llvm::Value*> multidim = { in EmitTilingKernel() local
4727 return IrArray::Index(multidim, dims_in_blocks, index_ty); in EmitTilingKernel()
4744 std::vector<llvm::Value*> elem_multi_index = block_coords.multidim(); in EmitTilingKernel()
4907 Permute(index.multidim(), {0, 2, 1}), in EmitHlo021Tile()
5763 index.multidim()[dim], in EmitElementForInputFusibleSlices()
5766 index.multidim()[dim], in EmitElementForInputFusibleSlices()
[all …]
/external/angle/third_party/vulkan-deps/glslang/src/Test/baseResults/
Dhlsl.array.multidim.frag.out1 hlsl.array.multidim.frag
/external/deqp-deps/glslang/Test/baseResults/
Dhlsl.array.multidim.frag.out1 hlsl.array.multidim.frag
/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.cc1818 std::vector<llvm::Value*> operand_multi_index = source_index.multidim(); in EmitElementalConcatenate()
2113 std::vector<llvm::Value*> multi_index = padded_index.multidim(); in EmitElementalPad()
2332 std::vector<llvm::Value*> source_multi_index = target_index.multidim(); in MakeElementGenerator()
2453 IrArray::Index source_index(target_index.multidim(), in MakeElementGenerator()