Searched defs:embedded_computation (Results 1 - 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | buffer_liveness_test.cc | 402 auto embedded_computation = local
409 HloInstruction::CreateCall(vec_, {param}, embedded_computation));
|
| H A D | hlo_dataflow_analysis_test.cc | 1497 // embedded_computation:
1505 // %call = Call(embedded_computation, {%exp})
1516 auto embedded_computation = local
1527 HloInstruction::CreateCall(vector_shape_, {exp}, embedded_computation));
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | 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) { argument
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 198 milliseconds