Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for reduce_sum (0.28 sec)

  1. tensorflow/compiler/mlir/lite/ir/tfl_ops.td

      let summary = "Log softmax operator";
    
      let description = [{
        Computes element-wise log softmax activations with the following formula
    
          input - log(reduce_sum(exp(input), dim))
      }];
    
      let arguments = (ins TFL_TensorOf<[F32, QUI8, QI8, TFL_Quint8]>:$input);
    
      let results = (outs TFL_TensorOf<[F32, QUI8, QI8, TFL_Quint8]>:$output);
    
      let hasOptions = 1;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 19:09:08 UTC 2024
    - 186K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/quantize_model_test.py

            is greater than 10.
    
            Args:
              x: Input tensor.
    
            Returns:
              A map of: output key -> output result.
            """
            if math_ops.reduce_sum(x) > 10.0:
              out = math_ops.matmul(x, self.filters_0)
              out = nn_ops.bias_add(out, self.bias_0)
              return {'output': out}
    
            out = math_ops.matmul(x, self.filters_1)
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 03:36:50 UTC 2024
    - 235.6K bytes
    - Viewed (0)
Back to top