Search Options

Results per page
Sort
Preferred Languages
Advance

Results 21 - 30 of 83 for ArrayAttr (0.34 sec)

  1. tensorflow/compiler/mlir/quantization/tensorflow/cc/convert_asset_args_test.cc

      DictionaryAttr arg_attrs = main_func_op.getArgAttrDict(/*index=*/0);
    
      EXPECT_THAT(arg_attrs.get("tf_saved_model.bound_input"), IsNull());
    
      const ArrayRef<Attribute> index_path_attrs =
          mlir::cast<ArrayAttr>(arg_attrs.get("tf_saved_model.index_path"))
              .getValue();
      EXPECT_THAT(index_path_attrs, SizeIs(1));
      StringAttr index_path =
          mlir::dyn_cast_or_null<StringAttr>(index_path_attrs[0]);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 6.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/stablehlo/passes/insert_calibration_statistics_saver.cc

      // Set the insertion point right before the return op.
      builder.setInsertionPoint(&region.back().back());
    
      StringAttr output_file_path_attr = builder.getStringAttr(output_file_path);
      ArrayAttr ids_attr = builder.getStrArrayAttr(ids);
      ArrayAttr calibration_methods_attr =
          builder.getI32ArrayAttr(calibration_methods);
      builder.create<TF::CalibrationStatisticsSaverOp>(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 14 06:31:57 UTC 2024
    - 7.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/ir/tf_saved_model.cc

          >();
    }
    
    static LogicalResult VerifyIndexPath(Operation *op, NamedAttribute named_attr) {
      auto attr = mlir::dyn_cast<ArrayAttr>(named_attr.getValue());
      if (!attr) {
        return op->emitError() << "'" << kTfSavedModelIndexPathAttr
                               << "' attribute should be an ArrayAttr";
      }
      for (auto element : attr) {
        if (mlir::isa<StringAttr>(element)) {
          continue;
        }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 23.5K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/decompose_reduce_dataset.cc

        const int func_input_size = func_inputs.size();
        const int argument_size =
            reduce_dataset->getAttrOfType<ArrayAttr>("Targuments").size();
        const int state_size =
            reduce_dataset->getAttrOfType<ArrayAttr>("Tstate").size();
        const int dataset_input_size = func_input_size - state_size - argument_size;
    
        SmallVector<Type, 2> dataset_types;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 14K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tf2xla/transforms/utils.cc

    }
    
    ConstantOp GetScalarNegZeroOfType(Type ty, Location loc, OpBuilder* builder) {
      return builder->create<ConstantOp>(loc, hlo::getScalarNegZeroOfType(ty));
    }
    
    DenseIntElementsAttr GetI64ElementsAttr(ArrayAttr attr) {
      RankedTensorType ty =
          RankedTensorType::get(static_cast<int64_t>(attr.size()),
                                IntegerType::get(attr.getContext(), 64));
      return DenseIntElementsAttr::get(ty, attr.getValue());
    }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 06 19:12:29 UTC 2023
    - 1.8K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/transforms/fused_kernel_matcher.cc

        // contraction, with two additions: the list of ops which have been fused
        // together; epsilon (only with FusedBatchNorm).
        std::vector<NamedAttribute> attrs = contraction->getAttrs();
        ArrayAttr fused_ops_attr = ArrayAttr::get(context, fused_ops);
        attrs.push_back(
            NamedAttribute(StringAttr::get(context, "fused_ops"), fused_ops_attr));
        // Epsilon is used only in fusions with the FusedBatchNorm op, so we zero it
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 14.9K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_main_function.cc

        main_func.setArgAttr(
            i, kTfSavedModelIndexPathAttr,
            ArrayAttr::get(context, {StringAttr::get(context, input_names[i])}));
      }
    
      const int num_results = main_func.getNumResults();
      for (int i = 0; i < num_results; ++i) {
        main_func.setResultAttr(
            i, kTfSavedModelIndexPathAttr,
            ArrayAttr::get(context, {StringAttr::get(context, output_names[i])}));
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 16.5K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/utils/translate_utils.cc

          version_attr.get("min_consumer"));
      if (min_consumer) versions->set_min_consumer(min_consumer.getInt());
    
      auto bad_consumers = mlir::dyn_cast_or_null<mlir::ArrayAttr>(
          version_attr.get("bad_consumers"));
      if (!bad_consumers) return mlir::success();
    
      for (auto bad_consumer : bad_consumers) {
        auto bad_consumer_int_attr =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Apr 26 09:37:10 UTC 2024
    - 3.2K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/ir/tf_op_base.td

    // attribute.
    def TF_SymbolRefArrayAttr :
      TypedArrayAttrBase<SymbolRefAttr, "tensorflow symbol ref array attribute"> {
      let constBuilderCall = "::mlir::ArrayAttr::get($_builder.getContext(), $0)";
    }
    
    //===----------------------------------------------------------------------===//
    // TensorFlow type definitions
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Oct 30 06:52:55 UTC 2023
    - 30.5K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/quantization/stablehlo/passes/replace_stablehlo_ops_in_main_function_with_xla_call_module_ops.cc

        shape_attrs.push_back(
            tf_type::ShapeAttr::get(ctx, mlir::cast<ShapedType>(result_type)));
      }
      const auto empty_array_attr = ArrayAttr::get(ctx, {});
      // TODO: b/310291615 - find a better way for platform support.
      const auto platforms = ArrayAttr::get(
          ctx,
          {StringAttr::get(ctx, kPlatformCpu), StringAttr::get(ctx, kPlatformTpu)});
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 21K bytes
    - Viewed (0)
Back to top