Home
last modified time | relevance | path

Searched defs:multidim (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
H A Dir_array.cc39 IrArray::Index::Index(absl::Span<llvm::Value* const> multidim, in Index()
47 void IrArray::Index::Delinearize(std::vector<llvm::Value*>* multidim, in Delinearize()
77 void IrArray::Index::Delinearize(std::vector<llvm::Value*>* multidim, in Delinearize()
121 absl::Span<llvm::Value* const> multidim, in Index()
161 IrArray::Index::Index(absl::Span<llvm::Value* const> multidim, in Index()
167 IrArray::Index::Index(absl::Span<llvm::Value* const> multidim, in Index()
H A Dir_array.h104 const std::vector<llvm::Value*>& multidim() const { return multidim_; } in multidim() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dparallel_loop_emitter.cc160 std::vector<llvm::Value*> multidim(shape_.rank(), nullptr); in EmitLogicalIndexAndSetExitBasicBlock() local
261 std::vector<llvm::Value*> multidim(shape_.rank(), nullptr); in EmitIndexAndSetExitBasicBlock() local
274 std::vector<llvm::Value*> multidim(shape_.rank(), nullptr); in EmitIndexAndSetExitBasicBlock() local
H A Dir_emitter_unnested.cc3488 auto multidim = normalized_shape_index.multidim(); in GetUnnormalizedIndex() local
3497 auto multidim = normalized_shape_index.multidim(); in GetUnnormalizedIndex() local
4129 std::vector<llvm::Value*> multidim = { in EmitTilingKernel() local
/aosp_15_r20/external/libbpf/src/
H A Dbtf_dump.c1502 bool multidim; in btf_dump_emit_type_chain() local