- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for AsProtoTensorContent (0.36 sec)
-
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) -
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)