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

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
ir_emitter_nested.cc 38 const HloComputation& nested_computation,
43 EmitBasePointersForNestedComputation(nested_computation, &io_hlos);
47 const HloComputation& nested_computation,
52 nested_computation.parameter_instructions()) {
62 const HloInstruction* root = nested_computation.root_instruction();
81 nested_computation.name()))), // The name of the function.
102 for (const auto* hlo : nested_computation.instructions()) {
104 hlo != nested_computation.root_instruction()) {
ir_emitter_nested.h 46 const HloComputation& nested_computation,
63 const HloComputation& nested_computation,
ir_emitter.h 142 const HloComputation& nested_computation,
145 // Emits an atomic operation that implements `nested_computation` in the
150 const HloComputation& nested_computation, llvm::Value* output_address,
203 // Emits an atomic operation that implements `nested_computation` in the
208 const HloComputation& nested_computation, llvm::Type* element_ir_type);
ir_emitter.cc 157 const HloComputation& nested_computation,
159 TF_RET_CHECK(nested_computation.num_parameters() > 0);
161 computation_to_ir_function_[&nested_computation];
163 IrEmitterNested ir_emitter_nested(hlo_module_config_, nested_computation,
166 nested_computation.root_instruction()->Accept(&ir_emitter_nested));
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_module.cc 458 HloComputation* nested_computation = AddEmbeddedComputation( local
461 output->shape(), arguments, nested_computation));
468 VLOG(2) << "to " << nested_computation->ToString();
  /external/tensorflow/tensorflow/compiler/xla/tests/
fusion_test.cc 657 auto nested_computation = local
662 nested_computation));
    [all...]

Completed in 259 milliseconds