Search Options

Results per page
Sort
Preferred Languages
Advance

Results 21 - 30 of 181 for SmallVector (0.17 sec)

  1. tensorflow/compiler/mlir/tfrt/ir/tfrt_fallback_common.cc

      mlir::IntegerAttr op_key;
      mlir::IntegerAttr cost;
      mlir::StringAttr device;
      mlir::StringAttr op_name;
      mlir::SymbolRefAttr f;
      llvm::SmallVector<mlir::OpAsmParser::UnresolvedOperand, 4> in_chains;
      llvm::SmallVector<mlir::OpAsmParser::UnresolvedOperand, 4> operands;
      mlir::NamedAttrList op_attrs;
      mlir::NamedAttrList op_func_attrs;
      auto loc = parser.getNameLoc();
    
      if (options.has_chain &&
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 5.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/utils/dynamic_shape_utils.cc

    #include "llvm/ADT/SmallVector.h"
    
    namespace tensorflow {
    
    llvm::SmallVector<int64_t> ConvertTFShapeToMlir(
        llvm::ArrayRef<int64_t> shapes) {
      return llvm::to_vector(llvm::map_range(shapes, [](int64_t shape) {
        return shape == kTFDynamicSize ? mlir::ShapedType::kDynamic : shape;
      }));
    }
    
    llvm::SmallVector<int64_t> ConvertMlirShapeToTF(
        llvm::ArrayRef<int64_t> shapes) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Nov 21 16:21:18 UTC 2022
    - 1.6K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/transforms/legalize_tensorlist.td

    def Size1InputRange : NativeCodeCall<
      "SmallVector<Value, 1>{$0}">;
    
    def Size2InputRange : NativeCodeCall<
      "SmallVector<Value, 2>{$0, $1}">;
    
    def Size3InputRange : NativeCodeCall<
      "SmallVector<Value, 3>{$0, $1, $2}">;
    
    def EmptyCustomOptions : NativeCodeCall<
      "TFL::ConstBytesAttr::get($_builder.getContext(), \"\")">;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Oct 18 07:12:51 UTC 2023
    - 3.4K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/quantization/stablehlo/passes/insert_calibration_statistics_saver.cc

    void FindCustomAggregatorOps(
        Region& region,
        const std::unordered_set<std::string>& aggregator_ops_to_ignore,
        SmallVector<Value>& statistics_outputs, SmallVector<StringRef>& ids,
        SmallVector<int32_t>& calibration_methods) {
      for (auto op : region.getOps<TF::CustomAggregatorOp>()) {
        if (aggregator_ops_to_ignore.count(op.getId().str())) continue;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 14 06:31:57 UTC 2024
    - 7.2K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/quantization/common/lift_as_function_call.h

    #define TENSORFLOW_COMPILER_MLIR_QUANTIZATION_COMMON_LIFT_AS_FUNCTION_CALL_H_
    
    #include "absl/base/nullability.h"
    #include "absl/status/statusor.h"
    #include "llvm/ADT/SmallVector.h"
    #include "mlir/Bytecode/BytecodeOpInterface.h"  // from @llvm-project
    #include "mlir/Dialect/Func/IR/FuncOps.h"  // from @llvm-project
    #include "mlir/IR/Attributes.h"  // from @llvm-project
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 10 04:07:09 UTC 2024
    - 5.5K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/transforms/xla_inline_device_ops.cc

    limitations under the License.
    ==============================================================================*/
    
    // This pass remove Cluster ops by inlining Cluster ops.
    
    #include "llvm/ADT/SmallVector.h"
    #include "tensorflow/compiler/mlir/tensorflow/ir/tf_device.h"
    #include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.h"
    #include "tensorflow/compiler/mlir/tensorflow/ir/tf_types.h"
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Oct 04 00:59:46 UTC 2022
    - 2.6K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/transforms/tpu_identity_pruning.cc

              if (func && reachable_funcs.insert(func).second)
                funcs_to_visit.push_back(func);
            });
          };
    
      SmallVector<func::FuncOp, 4> funcs_to_visit;
      collect_reachable_funcs(region, funcs_to_visit);
    
      while (!funcs_to_visit.empty()) {
        SmallVector<func::FuncOp, 4> new_funcs_to_visit;
        for (func::FuncOp func_to_visit : funcs_to_visit) {
          if (!func_to_visit.getCallableRegion()) continue;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Oct 05 23:50:19 UTC 2022
    - 4.1K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/lite/stablehlo/transforms/optimize_layout.cc

        ArrayRef<int64_t> transpose_perm = transpose_op.getPermutation();
        SmallVector<int64_t> new_padding_low =
            InvertPermutationToVector(pad_op.getEdgePaddingLow(), transpose_perm);
        SmallVector<int64_t> new_padding_high =
            InvertPermutationToVector(pad_op.getEdgePaddingHigh(), transpose_perm);
        SmallVector<int64_t> new_padding_interrier =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 21:59:06 UTC 2024
    - 8.6K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/ir/tf_op_interfaces.td

            return $_op.getDataFormat();
        }]>,
        InterfaceMethod<
          [{Returns indices of layout dependent arguments.}],
          "SmallVector<unsigned, 4>", "GetLayoutDependentArgs", (ins)
        >,
        InterfaceMethod<
          [{Returns indices of layout dependent results.}],
          "SmallVector<unsigned, 4>", "GetLayoutDependentResults", (ins)
        >,
        InterfaceMethod<
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Nov 30 19:07:07 UTC 2022
    - 6.2K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/utils/dynamic_shape_utils.h

    #define TENSORFLOW_COMPILER_MLIR_TENSORFLOW_UTILS_DYNAMIC_SHAPE_UTILS_H_
    
    #include "mlir/IR/BuiltinTypes.h"  // from @llvm-project
    
    namespace tensorflow {
    
    llvm::SmallVector<int64_t> ConvertTFShapeToMlir(llvm::ArrayRef<int64_t> shapes);
    
    llvm::SmallVector<int64_t> ConvertMlirShapeToTF(llvm::ArrayRef<int64_t> shape);
    
    static constexpr int64_t kTFDynamicSize = -1;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Oct 18 11:40:17 UTC 2022
    - 1.4K bytes
    - Viewed (0)
Back to top