Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for GetBoolAttr (0.38 sec)

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

        Value output = rewriter.create<TF::ResizeBilinearOp>(
            conv_op.getLoc(), conv_op.getType(), conv_op.getLhs(),
            output_sizes_attr,
            /*align_corners=*/rewriter.getBoolAttr(align_corners),
            /*half_pixel_centers=*/rewriter.getBoolAttr(false));
        rewriter.replaceOp(conv_op, {output});
      }
    
      LogicalResult MatchToResizeBilinearOp(
          mhlo::ConvolutionOp conv_op, bool& align_corners,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 154.9K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/transforms/optimize.cc

            /*fused_activation_function=*/
            rewriter.getStringAttr(add_op.getFusedActivationFunction()),
            /*weights_format=*/rewriter.getStringAttr(fc_op.getWeightsFormat()),
            /*keep_num_dims=*/rewriter.getBoolAttr(fc_op.getKeepNumDims()),
            /*asymmetric_quantize_inputs=*/
            fc_op.getAsymmetricQuantizeInputsAttr());
        rewriter.replaceOp(add_op, fc.getOutput());
    
        return success();
      }
    };
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 00:40:15 UTC 2024
    - 102.3K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc

            (lhs_contracting_dims[0] == lhs_rank - 2 ? rewriter.getBoolAttr(true)
                                                     : rewriter.getBoolAttr(false));
        const int rhs_rank = rhs_value.getType().cast<TensorType>().getRank();
        const BoolAttr adj_y =
            (rhs_contracting_dims[0] == rhs_rank - 1 ? rewriter.getBoolAttr(true)
                                                     : rewriter.getBoolAttr(false));
    
        Value result = op.getResult();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 22 09:00:19 UTC 2024
    - 99.8K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/translate/import_model.cc

                              builder_.getBoolAttr(false));
        }
      }
    
      auto composite_control_flow_op = [&](const std::string& name) {
        result.name = mlir::OperationName(get_full_op_name(name), context_);
        bool stateless = absl::StartsWith(node_type_name, "Stateless");
        mlir::BoolAttr val = builder_.getBoolAttr(stateless);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 183.2K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc

      }
    
      // Shapes are known to be compatible.
      ReplaceTfOpWithNewOp<Ty>(rewriter, op, op.getX(), op.getY(),
                               rewriter.getBoolAttr(true));
      return success();
    }
    }  // namespace
    
    void EqualOp::getCanonicalizationPatterns(RewritePatternSet& results,
                                              MLIRContext* context) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 146.7K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc

      if (isa<TensorFlowDialect, BuiltinDialect>(element_type.getDialect()))
        return b.create<TF::CastOp>(loc, dst_type, input,
                                    /*truncate=*/b.getBoolAttr(false));
      return nullptr;
    }
    
    // Follow the use chain of TensorList and return true iff all elements written
    // to TensorList have same static shape. If all elements have same shape, assign
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Jun 08 07:28:49 UTC 2024
    - 134.1K bytes
    - Viewed (0)
Back to top