HomeSort by relevance Sort by last modified time
    Searched refs:LocalComputation (Results 1 - 4 of 4) 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...]
local_computation_builder.i 101 // Other SWIG object wrappers (e.g. of LocalComputation) are further
191 %typemap(out) StatusOr<xla::swig::LocalComputation*> {
196 $typemap(out, xla::swig::LocalComputation*)
868 %unignore xla::swig::LocalComputation;
869 %unignore xla::swig::LocalComputation::Compile;
870 %unignore xla::swig::LocalComputation::GetReturnValueShape;
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 195 milliseconds