Home
last modified time | relevance | path

Searched refs:quantized_input (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_concat_op.cc135 bool quantized_input) { in Compute() argument
164 if (quantized_input) { in Compute()
213 if (!quantized_input) { in Compute()
223 if (quantized_input) { in Compute()
540 bool quantized_input = in Compute() local
542 if (quantized_input) { in Compute()
576 mkl_input_shapes, quantized_input); in Compute()
781 if (quantized_input) { in Compute()
821 bool quantized_input) { in CallEigenVersion() argument
842 input_mins, input_maxes, quantized_input); in CallEigenVersion()
[all …]
/external/tensorflow/tensorflow/compiler/tests/
Dquantized_ops_test.py90 quantized_input = array_ops.quantize(transposed_input, 0.0, 255.0,
92 packed_input = self.pack_uint8_r2_to_uint32(quantized_input.output)
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dsvdf.h197 float* scaling_factors, int8_t* quantized_input, float* state, in EvalHybridSVDF() argument
218 input_data, batch_size, input_size, quantized_input, scaling_factors, in EvalHybridSVDF()
226 weights_feature_data, num_filters, input_size, quantized_input, in EvalHybridSVDF()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlower_tf.cc402 Value quantized_input = rewriter.create<ClipByValueOp>( in matchAndRewrite() local
405 quantized_input = rewriter.create<SubOp>(op.getLoc(), input_ty, in matchAndRewrite()
406 quantized_input, nudged_float_min); in matchAndRewrite()
408 quantized_input = rewriter.create<MulOp>(op.getLoc(), input_ty, in matchAndRewrite()
409 quantized_input, float_to_quant); in matchAndRewrite()
416 quantized_input = rewriter.create<AddV2Op>(op.getLoc(), input_ty, in matchAndRewrite()
417 quantized_input, half_val); in matchAndRewrite()
419 quantized_input = rewriter.create<FloorOp>(op.getLoc(), quantized_input); in matchAndRewrite()
423 quantized_input, quant_to_float); in matchAndRewrite()