HomeSort by relevance Sort by last modified time
    Searched refs:LocalComputationBuilder (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/python/
local_computation_builder.i 871 %unignore xla::swig::LocalComputationBuilder;
872 %unignore xla::swig::LocalComputationBuilder::LocalComputationBuilder;
873 %unignore xla::swig::LocalComputationBuilder::Build;
874 %unignore xla::swig::LocalComputationBuilder::SetOpMetadata;
875 %unignore xla::swig::LocalComputationBuilder::ClearOpMetadata;
876 %unignore xla::swig::LocalComputationBuilder::Parameter;
877 %unignore xla::swig::LocalComputationBuilder::GetShape;
878 %unignore xla::swig::LocalComputationBuilder::GetReturnValueShape;
879 %unignore xla::swig::LocalComputationBuilder::Infeed
    [all...]
local_computation_builder.cc 287 LocalComputationBuilder::LocalComputationBuilder(const string& computation_name)
290 void LocalComputationBuilder::SetOpMetadata(const OpMetadata& metadata) {
294 void LocalComputationBuilder::ClearOpMetadata() { builder_.ClearOpMetadata(); }
296 StatusOr<LocalComputation*> LocalComputationBuilder::Build() {
301 ComputationDataHandle LocalComputationBuilder::Parameter(int64 parameter_number,
307 std::unique_ptr<Shape> LocalComputationBuilder::GetShape(
312 StatusOr<Shape> LocalComputationBuilder::GetReturnValueShape() {
317 ComputationDataHandle LocalComputationBuilder::Infeed(const Shape& shape) {
321 void LocalComputationBuilder::Outfeed(const ComputationDataHandle& operand
    [all...]
local_computation_builder.h 98 // Wraps a Computation produced by a LocalComputationBuilder. The
126 class LocalComputationBuilder {
128 LocalComputationBuilder(const string& computation_name);
xla_client.py 459 self._client = c_api.LocalComputationBuilder(name.encode('utf8'))
    [all...]

Completed in 255 milliseconds