HomeSort by relevance Sort by last modified time
    Searched defs:embedded_computation (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
buffer_liveness_test.cc 402 auto embedded_computation = local
409 HloInstruction::CreateCall(vec_, {param}, embedded_computation));
    [all...]
hlo_dataflow_analysis_test.cc 1516 auto embedded_computation = local
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/tests/
map_test.cc 120 // "embedded_computation" on it, and then adds "n" to the result.
124 // embedded_computation --/ n --/
125 Computation CreateMapPlusN(const Computation& embedded_computation, float n) {
128 auto map = builder.Map({x}, embedded_computation, {});
354 auto embedded_computation = computation_status.ConsumeValueOrDie(); local
358 auto map_plus_1 = builder.Map({constant_vector}, embedded_computation, {0});
368 *op_request.mutable_computation() = embedded_computation.handle();
374 auto map_plus_2 = builder.Map({map_plus_1}, embedded_computation, {0});

Completed in 90 milliseconds