- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for input_float (0.09 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/calibrator/custom_aggregator_op.cc
// Calculate min/max statistics. const auto input_flat = input_tensor.flat<float>(); Tensor *min_output = nullptr, *max_output = nullptr; OP_REQUIRES_OK(context, context->allocate_output("min", {}, &min_output)); OP_REQUIRES_OK(context, context->allocate_output("max", {}, &max_output)); min_output->scalar<float>().device( context->template eigen_device<CPUDevice>()) = input_flat.minimum(); max_output->scalar<float>().device(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 01:09:50 UTC 2024 - 6.2K bytes - Viewed (0)