Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for AsProtoTensorContent (0.14 sec)

  1. tensorflow/compiler/mlir/quantization/stablehlo/utils/tf_type_utils_test.cc

      tensor.matrix<tsl::qint8>()(0, 1) = tsl::qint8(2);
      tensor.matrix<tsl::qint8>()(1, 0) = tsl::qint8(3);
      tensor.matrix<tsl::qint8>()(1, 1) = tsl::qint8(4);
    
      ::tensorflow::TensorProto tensor_proto;
      tensor.AsProtoTensorContent(&tensor_proto);
      return ::tensorflow::mangling_util::MangleTensor(tensor_proto);
    }
    
    std::string GetQint16Tensor() {
      ::tensorflow::Tensor tensor(::tensorflow::DT_QINT16, {2, 2});
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 9K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/tensorflow/debugging/dump_tensor_op.cc

      }
    
      void Compute(OpKernelContext* ctx) override {
        if (!enabled_) return;
    
        const Tensor& tensor_data = ctx->input(0);
    
        TensorProto tensor_proto;
        tensor_data.AsProtoTensorContent(&tensor_proto);
    
        OP_REQUIRES_OK(ctx, tensor_data_writer_->WriteRecord(
                                tensor_proto.SerializeAsString()));
      }
    
     private:
      bool enabled_;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 03:12:17 UTC 2024
    - 4.8K bytes
    - Viewed (0)
  3. 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)
  4. 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