Home
last modified time | relevance | path

Searched refs:ir_array (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
DBUILD37 ":ir_array",
92 name = "ir_array",
93 srcs = ["ir_array.cc"],
94 hdrs = ["ir_array.h"],
116 ":ir_array",
134 ":ir_array",
152 ":ir_array",
178 ":ir_array",
194 ":ir_array",
217 ":ir_array",
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dhlo_to_ir_bindings.cc198 llvm_ir::IrArray ir_array(base_ptr, in GetIrArray() local
210 ir_array.MarkInvariantOverWholeProgram(&module_->getContext()); in GetIrArray()
213 return ir_array; in GetIrArray()
Dir_emitter_unnested.h560 const Shape& output_shape, const llvm_ir::IrArray& ir_array,
DBUILD184 "//tensorflow/compiler/xla/service/llvm_ir:ir_array",
297 "//tensorflow/compiler/xla/service/llvm_ir:ir_array",
329 "//tensorflow/compiler/xla/service/llvm_ir:ir_array",
356 "//tensorflow/compiler/xla/service/llvm_ir:ir_array",
Dir_emitter_unnested.cc1800 auto ir_array = operand_arrays[i]; in EmitLoopFusionFromMlir() local
1803 [builder, ir_array](llvm_ir::IrArray::Index index) { in EmitLoopFusionFromMlir()
1804 return ir_array.EmitReadArrayElement(index, builder); in EmitLoopFusionFromMlir()
3503 llvm_ir::IrArray ir_array( in BuildKernelThunkForMlirImpl() local
3506 ir_array.MarkInvariantOverWholeProgram(&value->getContext()); in BuildKernelThunkForMlirImpl()
3509 ir_arrays->push_back(ir_array); in BuildKernelThunkForMlirImpl()
5630 auto ir_array = ir_arrays[i]; in EmitReductionFromOrToContiguousDimensions() local
5633 fused_operand, [this, ir_array, in EmitReductionFromOrToContiguousDimensions()
5635 return ir_array.EmitReadArrayElement(index, &b_, in EmitReductionFromOrToContiguousDimensions()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
DBUILD400 "//tensorflow/compiler/xla/service/llvm_ir:ir_array",
475 "//tensorflow/compiler/xla/service/llvm_ir:ir_array",
523 "//tensorflow/compiler/xla/service/llvm_ir:ir_array",
/external/tensorflow/tensorflow/compiler/xla/service/
DBUILD4295 "//tensorflow/compiler/xla/service/llvm_ir:ir_array",