Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for AsProtoTensorContent (0.15 sec)

  1. tensorflow/cc/tools/freeze_saved_model.cc

      const_node->set_name(variable_node.name());
      const_node->set_op("Const");
      (*const_node->mutable_attr())["dtype"] = variable_node.attr().at("dtype");
      variable_value.AsProtoTensorContent(
          (*const_node->mutable_attr())["value"].mutable_tensor());
    }
    
    // Converts a ReadVariableOp NodeDef to an Identity NodeDef.
    void ConvertReadVariableOpToIdentity(const NodeDef& node,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 11 08:05:36 UTC 2023
    - 11.9K bytes
    - Viewed (0)
  2. tensorflow/cc/framework/ops.h

          typedef T type;
        };
        // END_SKIP_DOXYGEN
    
        TensorProto AsTensorProto() {
          TensorProto tensor_proto;
          if (tensor.NumElements() > 1) {
            tensor.AsProtoTensorContent(&tensor_proto);
          } else {
            tensor.AsProtoField(&tensor_proto);
          }
          return tensor_proto;
        }
    
        Status status;
        Tensor tensor;
      };
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Apr 13 05:57:22 UTC 2024
    - 10.5K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/utils/convert_tensor.cc

      // representation.
      // * For scalars and short strings it may be marginally worse and a more
      //   intelligent decision could be made by caller.
      if (use_tensor_content)
        input_tensor.AsProtoTensorContent(&tensor_proto);
      else
        input_tensor.AsProtoField(&tensor_proto);
      return tensor_proto;
    }
    
    static std::string MangleTensor(const Tensor& tensor) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Apr 26 09:37:10 UTC 2024
    - 20.5K bytes
    - Viewed (0)
Back to top