Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dir_array.cc37 IrArray::Index::Index(absl::Span<llvm::Value* const> multidim, in Index()
45 void IrArray::Index::Delinearize(std::vector<llvm::Value*>* multidim, in Delinearize()
75 void IrArray::Index::Delinearize(std::vector<llvm::Value*>* multidim, in Delinearize()
130 IrArray::Index::Index(absl::Span<llvm::Value* const> multidim, in Index()
136 IrArray::Index::Index(absl::Span<llvm::Value* const> multidim, in Index()
Dir_array.h98 const std::vector<llvm::Value*>& multidim() const { return multidim_; } in multidim() function
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc4127 auto multidim = normalized_shape_index.multidim(); in GetUnnormalizedIndex() local
4723 std::vector<llvm::Value*> multidim = { in EmitTilingKernel() local