Search Options

Results per page
Sort
Preferred Languages
Advance

Results 31 - 40 of 46 for TypeAttr (0.11 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/lift_variables.cc

        builder.create<tf_saved_model::GlobalTensorOp>(
            NameLoc::get(builder.getStringAttr(name.str())),
            builder.getStringAttr(name), tensor_attr,
            TypeAttr::get(tensor_attr.getType()), builder.getUnitAttr());
      }
    
      return success();
    }
    
    }  // namespace
    
    LogicalResult LiftVariables(ModuleOp module, Session* session) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 23 09:05:47 UTC 2024
    - 7.9K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/ir/tf_saved_model_ops.td

        sense of `tf.TensorShape` compatibility. And the element types must match.
      }];
    
      let arguments = (ins
        StrAttr:$sym_name,
        OptionalAttr<ElementsAttr>:$value,
        TypeAttr:$type,
        UnitAttr:$is_mutable
      );
    
      let hasVerifier = 1;
    }
    
    def TfSavedModel_SessionInitializerOp: TfSavedModel_Op<"session_initializer"> {
      let summary = "Initializes TensorFlow session state.";
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Oct 31 08:49:35 UTC 2023
    - 7.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/ir/tfl_ops.td

        // determined while going through quantization passes.
        OptionalAttr<TypeAttr>:$input_to_input_intermediate,
        OptionalAttr<TypeAttr>:$input_to_forget_intermediate,
        OptionalAttr<TypeAttr>:$input_to_cell_intermediate,
        OptionalAttr<TypeAttr>:$input_to_output_intermediate,
        OptionalAttr<TypeAttr>:$effective_hidden_scale_intermediate
      );
    
      let results = (outs AnyTensor:$output);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 19:09:08 UTC 2024
    - 186K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/ir/tf_op_base.td

              "mlir::OperandElementTypeIterator(values.end())};",
      [{
        ArrayAttr::get($_ctxt,
        [&]() {
          llvm::SmallVector<Attribute, 4> ret;
          for (auto t : $_self)
            ret.push_back(TypeAttr::get(t));
          return ret;
        }())
      }]
    >;
    
    // A derived attribute that returns the shapes of the tensors in the actual
    // value pack that corresponds to the `idx`-th ODS-declared variadic operand.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Oct 30 06:52:55 UTC 2023
    - 30.5K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/decompose_reduce_dataset.cc

      llvm::SmallVector<Attribute, 2> type_attrs;
      for (Type type : dataset_types) {
        shape_attrs.push_back(
            TF::ShapeAttr::get(builder.getContext(), mlir::cast<ShapedType>(type)));
        type_attrs.push_back(TypeAttr::get(getElementTypeOrSelf(type)));
      }
    
      auto anonymous_iterator = builder.create<AnonymousIteratorV3Op>(
          reduce_dataset.getLoc(),
          RankedTensorType::get({}, builder.getType<ResourceType>()),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 14K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/transforms/fused_kernel_matcher.cc

          // Here TArgs types do not include types of the first two parameters,
          // i.e. the convolution input and the filter. TArgs are parameters for
          // the extras like the bias etc.
          auto attr = TypeAttr::get(getElementTypeOrSelf(contraction.getType()));
          SmallVector<Attribute, 4> targs_values(operands.size() - 2, attr);
    
          ArrayAttr targs_attr = ArrayAttr::get(context, targs_values);
          attrs.push_back(
    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/tensorflow/transforms/tfg-to-tfe.cc

                res_types, res_attrs)))
          return failure();
    
        // Update the function type which has excluded the control args.
        func->setAttr("function_type", TypeAttr::get(rewriter.getFunctionType(
                                           arg_types, res_types)));
    
        // Update arg/result attributes.
        func.setAllArgAttrs(arg_attrs);
        func.setAllResultAttrs(res_attrs);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 21.2K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/quantization/tensorflow/passes/prepare_quantize.cc

            auto min_max = GetMinMaxValuesForArgument(func_name, i);
            // The input min/max or mean/std are not specified, then skip.
            if (!min_max.first.has_value() || !min_max.second.has_value()) return;
    
            TypeAttr params = quant::GetQuantizedTypeAttr(
                builder, input_type, builder.getF64FloatAttr(min_max.first.value()),
                builder.getF64FloatAttr(min_max.second.value()),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 17.2K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/lite/transforms/prepare_quantize.cc

            auto min_max = GetMinMaxValuesForArgument(func_name, i);
            // The input min/max or mean/std are not specified, then skip.
            if (!min_max.first.has_value() || !min_max.second.has_value()) return;
    
            TypeAttr params = quant::GetQuantizedTypeAttr(
                builder, input_type, builder.getF64FloatAttr(min_max.first.value()),
                builder.getF64FloatAttr(min_max.second.value()),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 17.6K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/lite/experimental/tac/transforms/device_transform_patterns.cc

                                       mean_op.getAxis(), mean_op.getKeepDims());
    
      // Insert a requant op.
      rewriter.replaceOpWithNewOp<TFL::QuantizeOp>(
          mean_op, output_type, new_mean_op, mlir::TypeAttr::get(output_type));
      return success();
    }
    
    }  // namespace tac
    }  // namespace TFL
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 25.4K bytes
    - Viewed (0)
Back to top