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

  /external/tensorflow/tensorflow/compiler/xla/python/
local_computation_builder.h 74 // LocalComputation. The Execute method forwards to that of the
102 class LocalComputation {
104 LocalComputation(Computation computation);
133 // Returns an owned LocalComputation to the caller on success.
134 StatusOr<LocalComputation*> Build();
187 const ComputationDataHandle& operand, const LocalComputation& select,
192 const ComputationDataHandle& init_value, const LocalComputation& scatter);
219 const LocalComputation& local_computation,
231 const LocalComputation& local_computation,
238 const LocalComputation& local_computation
    [all...]
local_computation_builder.cc 254 LocalComputation::LocalComputation(Computation computation)
257 StatusOr<CompiledLocalComputation*> LocalComputation::Compile(
277 const Computation& LocalComputation::computation() const {
281 StatusOr<Shape> LocalComputation::GetReturnValueShape() const {
296 StatusOr<LocalComputation*> LocalComputationBuilder::Build() {
298 return new LocalComputation(std::move(computation));
392 const ComputationDataHandle& operand, const LocalComputation& select,
397 const ComputationDataHandle& init_value, const LocalComputation& scatter) {
442 const LocalComputation& local_computation
    [all...]
xla_client.py 34 # whereas method names of ComputationBuilder and LocalComputation are
349 class LocalComputation(object):
352 A LocalComputation can be executed if it is compiled. Otherwise, it
366 assert isinstance(c_local_computation, c_api.LocalComputation)
401 return LocalComputation(
447 LocalComputation, which in turn can be compiled into a
463 return LocalComputation(self._client.Build(), is_compiled=False)
    [all...]

Completed in 169 milliseconds