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

  /external/tensorflow/tensorflow/compiler/xla/python/
local_computation_builder.h 98 // Wraps a Computation produced by a LocalComputationBuilder. The
126 class LocalComputationBuilder {
128 LocalComputationBuilder(const string& computation_name);
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...]

Completed in 104 milliseconds