Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dfused_ir_emitter.cc85 .EmitReadArrayElement(index, ir_builder_); in HandleConstant()
113 .EmitReadArrayElement(index, ir_builder_); in HandleGetTupleElement()
122 .EmitReadArrayElement(index, ir_builder_); in HandleParameter()
Dops.cc100 return start_indices_array.EmitReadArrayElement(index, ir_builder); in EmitDynamicUpdateSliceInPlace()
103 return update_array.EmitReadArrayElement(index, ir_builder); in EmitDynamicUpdateSliceInPlace()
Dir_array.h209 llvm::Value* EmitReadArrayElement(const Index& index,
Dir_array.cc281 llvm::Value* IrArray::EmitReadArrayElement(const Index& index, in EmitReadArrayElement() function in xla::llvm_ir::IrArray
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.cc74 .EmitReadArrayElement(index, &ir_builder_); in DefaultAction()
454 lhs_array.EmitReadArrayElement(/*index=*/{}, &ir_builder_); in HandleDot()
456 rhs_array.EmitReadArrayElement(/*index=*/{}, &ir_builder_); in HandleDot()
545 lhs_array.EmitReadArrayElement(lhs_index, &ir_builder_); in HandleDot()
547 rhs_array.EmitReadArrayElement(rhs_index, &ir_builder_); in HandleDot()
728 .EmitReadArrayElement(index, &ir_builder_); in HandleRng()
Dir_emitter_unnested.cc864 input.EmitReadArrayElement(index, builder, "input_element"), in EmitTranspose021Tiled()
1683 .EmitReadArrayElement(index, &ir_builder_); in HandleReduce()
1687 .EmitReadArrayElement(index, &ir_builder_); in HandleReduce()
1851 operand_array.EmitReadArrayElement(operand_index, &ir_builder_); in HandleSelectAndScatter()
2292 .EmitReadArrayElement(index, &ir_builder_); in EmitInitializer()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter.cc816 lhs_array_.EmitReadArrayElement(lhs_index, ir_builder_); in Emit()
818 rhs_array_.EmitReadArrayElement(rhs_index, ir_builder_); in Emit()
881 lhs_array_.EmitReadArrayElement(/*index=*/{}, ir_builder_); in EmitScalarDot()
883 rhs_array_.EmitReadArrayElement(/*index=*/{}, ir_builder_); in EmitScalarDot()
Dir_emitter.cc741 operand_array.EmitReadArrayElement(operand_index, &ir_builder_); in HandleSelectAndScatter()
1133 input_array.EmitReadArrayElement(input_index, &ir_builder_), in HandleConvolution()
1134 kernel_array.EmitReadArrayElement(kernel_index, &ir_builder_)); in HandleConvolution()
2029 operand_array.EmitReadArrayElement(operand_index, &ir_builder_); in HandlePad()
2936 return GetIrArrayFor(operand).EmitReadArrayElement(index, &ir_builder_); in DefaultAction()