Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for AsProtoTensorContent (0.17 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)
Back to top