Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for isContained (0.14 sec)

  1. tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc

            llvm::to_vector(op.getBroadcastDimensions());
    
        while (input_dims.size() < output_rank) {
          int32_t dim_to_expand = 0;
          for (int32_t i = 0; i < output_rank; ++i) {
            if (!llvm::is_contained(input_dims, i)) {
              dim_to_expand = i;
              break;
            }
          }
    
          TensorType dim_type = input_type.cloneWith({static_cast<int64_t>(1)},
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 22 09:00:19 UTC 2024
    - 99.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc

        }
      }
    
      // If strides are constants, verify that none of the element is zero.
      DenseIntElementsAttr strides;
      if (matchPattern(op.getStrides(), m_Constant(&strides))) {
        if (llvm::is_contained(strides.getValues<APInt>(), 0))
          return op.emitOpError("requires non-zero strides");
      }
    
      // Use bit compares to ensure ellipsis_mask is 0 or a power of 2, i.e. there
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 170.8K bytes
    - Viewed (0)
Back to top