HomeSort by relevance Sort by last modified time
    Searched refs:EmitReadArrayElement (Results 1 - 10 of 10) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
cpu_noalias_test.cc 94 param_x_array.EmitReadArrayElement(zero2D, &b)
104 concat1_array.EmitReadArrayElement(zero2D, &b)
114 concat2_array.EmitReadArrayElement(zero2D, &b)
  /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
fused_ir_emitter.cc 98 .EmitReadArrayElement(index, b_);
136 .EmitReadArrayElement(index, b_);
164 .EmitReadArrayElement(index, b_);
dynamic_update_slice_util.cc 120 return operand_arrays[2 + index].EmitReadArrayElement(
124 return update_array.EmitReadArrayElement(index, b);
ir_array.h 244 llvm::Value* EmitReadArrayElement(const Index& index, llvm::IRBuilder<>* b,
sort_util.cc 196 auto value = params[i].EmitReadArrayElement(keys_index, b);
ir_array.cc 391 llvm::Value* IrArray::EmitReadArrayElement(const Index& index,
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
ir_emitter.cc 72 return GetIrArray(*operand, *hlo).EmitReadArrayElement(index, &b_);
486 lhs_array.EmitReadArrayElement(/*index=*/element_index, &b_);
488 rhs_array.EmitReadArrayElement(/*index=*/element_index, &b_);
581 llvm::Value* lhs_element = lhs_array.EmitReadArrayElement(lhs_index, &b_);
584 llvm::Value* rhs_element = rhs_array.EmitReadArrayElement(rhs_index, &b_);
    [all...]
ir_emitter_unnested.cc     [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
dot_op_emitter.cc 490 llvm::Value* lhs_element = lhs_array_.EmitReadArrayElement(lhs_index, b_);
491 llvm::Value* rhs_element = rhs_array_.EmitReadArrayElement(rhs_index, b_);
553 lhs_array_.EmitReadArrayElement(/*index=*/element_index, b_);
555 rhs_array_.EmitReadArrayElement(/*index=*/element_index, b_);
    [all...]
ir_emitter.cc     [all...]

Completed in 222 milliseconds