Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for KnownShape (0.09 sec)

  1. tensorflow/compiler/jit/encapsulate_subgraphs_pass.cc

      // Don't set any value field in the proto, since it is only going to be used
      // for shape inference.
    
      GraphDefBuilder::Options options(graph_out, /*status=*/nullptr);
      NodeBuilder node_builder(options.GetNameForOp("KnownShape"), "Const",
                               options.op_registry());
      node_builder.Attr("dtype", data_type).Attr("value", dummy_proto);
      Node* node = options.FinalizeBuilder(&node_builder);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 51K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/encapsulate_subgraphs_pass_test.cc

        value.mutable_tensor_shape()->add_dim()->set_size(dim);
      }
      return opts.WithAttr("value", value)
          .WithAttr("dtype", dtype)
          .FinalizeBuilder(&node_builder);
    }
    
    Node* KnownShape(absl::Span<const int> shape,
                     const GraphDefBuilder::Options& opts) {
      return KnownShapeBase(DT_FLOAT, shape, opts);
    }
    
    Node* KeyPlaceholderShape(const GraphDefBuilder::Options& opts) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 113.3K bytes
    - Viewed (0)
Back to top