Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for mutable_tensor_shape (0.26 sec)

  1. tensorflow/compiler/jit/encapsulate_subgraphs_pass_test.cc

      NodeBuilder node_builder(opts.GetNameForOp("Const"), "Const",
                               opts.op_registry());
      TensorProto value;
      value.set_dtype(dtype);
      for (int dim : shape) {
        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,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 113.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/translate/import_model.cc

      inputs.reserve(assets.size());
      for (const auto& asset : assets) {
        TensorInfo tensor_info;
        tensor_info.set_name(asset.tensor_name);
        tensor_info.set_dtype(DT_STRING);
        tensor_info.mutable_tensor_shape();
        inputs.push_back({asset.tensor_name, tensor_info});
      }
    
      std::unordered_map<std::string, std::string> tf_name_to_mlir_name;
      TF_ASSIGN_OR_RETURN(auto sub_module,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 183.2K bytes
    - Viewed (0)
Back to top