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

  /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
tuple_ops.cc 39 ir_builder->CreateLoad(pred.GetBasePointer(), "load_predicate_value");
64 ir_builder->CreateInBoundsGEP(select.GetBasePointer(), element_index);
81 tuple.GetBasePointer(),
fused_ir_emitter.cc 128 parameter_arrays_[parameter->parameter_number()].GetBasePointer()));
ir_array.h 180 llvm::Value* GetBasePointer() const { return base_ptr_; }
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
ir_emitter.h 119 // A convenient helper for calling HloToIrBindings::GetBasePointer.
120 llvm::Value* GetBasePointer(const HloInstruction& inst) const {
121 return bindings_.GetBasePointer(inst);
hlo_to_ir_bindings.h 73 llvm::Value* GetBasePointer(const HloInstruction& hlo,
ir_emitter.cc 107 bindings_.BindHloToIrValue(*bitcast, GetBasePointer(*operand));
121 /*alignment=*/1, GetBasePointer(*operand), &ir_builder_, module_));
149 base_ptrs.push_back(GetBasePointer(*operand));
430 GetBasePointer(*on_true),
431 GetBasePointer(*on_false), &ir_builder_, module_);
640 ir_builder_.CreateLoad(GetBasePointer(*init_value)),
703 operand_addresses.push_back(GetBasePointer(*operand));
706 GetBasePointer(*call));
    [all...]
hlo_to_ir_bindings.cc 199 llvm::Value* base_ptr = GetBasePointer(hlo, shape_index);
ir_emitter_unnested.cc     [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
dot_op_emitter.cc 620 llvm::Value* result_op = target_array_.GetBasePointer();
622 swap_operands ? rhs_array_.GetBasePointer() : lhs_array_.GetBasePointer();
624 swap_operands ? lhs_array_.GetBasePointer() : rhs_array_.GetBasePointer();
658 addend_array_ ? addend_array_->GetBasePointer() : nullptr, result_op,
681 addend_array_ ? addend_array_->GetBasePointer() : nullptr, result_op,
    [all...]
ir_emitter.cc     [all...]

Completed in 160 milliseconds