Searched refs:local_computation (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.cc | 608 LocalOp ComputationBuilder::Call(const Computation& local_computation, in Call() argument
615 return xla::Call(&builder_, local_computation.computation(), xla_ops); in Call()
644 const Computation& local_computation, in Map() argument
652 return xla::Map(&builder_, xla_ops, local_computation.computation(), in Map()
658 const Computation& local_computation, in Reduce() argument
661 local_computation.computation(), dimensions_to_reduce); in Reduce()
666 const Computation& local_computation, in ReduceWindowWithGeneralPadding() argument
673 operand.op(), init_value.op(), local_computation.computation(), in ReduceWindowWithGeneralPadding()
|
| D | local_computation_builder.h | 308 LocalOp Call(const Computation& local_computation,
323 const Computation& local_computation,
327 const Computation& local_computation,
332 const Computation& local_computation,
|