Search Options

Results per page
Sort
Preferred Languages
Advance

Results 41 - 50 of 345 for SmallVector (0.28 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/tpu_rewrite_device_util_test.cc

        parsed_devices->push_back(parsed_name);
      }
      return true;
    }
    
    using DeviceNames = llvm::SmallVector<std::string, 8>;
    
    struct ParameterizedDeviceSetTest
        : ::testing::TestWithParam<std::tuple<DeviceNames, std::string>> {};
    
    TEST_P(ParameterizedDeviceSetTest, BadDeviceSet) {
      llvm::SmallVector<Device, 8> devices;
      ASSERT_TRUE(DeviceNamesToParsedNames(std::get<0>(GetParam()), &devices));
      std::string topology_attr;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Apr 26 09:37:10 UTC 2024
    - 46.8K bytes
    - Viewed (0)
  3. 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)
  4. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc

        // Compute ConvDimensionNumbers, dilation, and padding.
        SmallVector<int64_t, num_spatial_dims> spatial_dims;
        SmallVector<int64_t, num_spatial_dims> kernel_spatial_dims;
        SmallVector<int64_t, num_spatial_dims> rhs_dilation;
        SmallVector<int64_t, num_spatial_dims * 2> paddings;
        SmallVector<int64_t, num_spatial_dims> window_strides;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 291.8K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/utils/const_tensor_utils.cc

    // bytes that represent host endianness values.
    // The read_size parameter is present to allow reading both float16 and float32
    // without a case split.
    template <typename T>
    llvm::SmallVector<mlir::APInt> ReadAsHostEndian(ArrayRef<uint8_t> bytes) {
      llvm::SmallVector<mlir::APInt> ret;
      size_t read_size = sizeof(T);
      int bytes_len = bytes.size();
      assert(bytes_len % read_size == 0);
    
      int elem_count = bytes_len / read_size;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 07 23:04:40 UTC 2024
    - 16.6K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/transforms/tpu_space_to_depth_pass.cc

      MLIRContext* context = backprop.getContext();
      SmallVector<int64_t, 4> values = {1, 1, 1, 1};
      auto attrs = llvm::map_range(values, [context](int64_t v) -> Attribute {
        return IntegerAttr::get(IntegerType::get(context, 64), APInt(64, v));
      });
      auto strides = ArrayAttr::get(context, llvm::to_vector<4>(attrs));
    
      // new result type.
      SmallVector<int64_t, 4> new_shape(new_filter_shape.begin(),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 29.3K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/quantization/stablehlo/passes/quantization_patterns.h

     private:
      // Collects all candidate ops for quantization, which are the
      // `dequantize_op`'s users.
      FailureOr<SmallVector<Operation*>> CollectCandidateOps(
          DequantizeOpT dequantize_op) const {
        auto users = dequantize_op->getResult(0).getUsers();
        return SmallVector<Operation*>(users.begin(), users.end());
      }
    
      // Collects all candidate ops for quantization, which is the operand of
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 10.9K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/quantization/common/lift_as_function_call.cc

      SmallVector<NamedAttribute> attributes;
      return LiftAsFunctionCall(builder, location, call_op_type, func_name,
                                arguments, results, attributes);
    }
    
    SmallVector<Value> AppendToVector(const ArrayRef<Value> arguments,
                                      Value append) {
      SmallVector<Value> ret(arguments);
      ret.push_back(append);
      return ret;
    }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 21.8K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/quantization/stablehlo/passes/replace_stablehlo_ops_in_main_function_with_xla_call_module_ops.cc

                               const SetVector<Operation*>& ops_to_add,
                               SmallVector<Operation*>& all_descendants) {
      if (!op) {
        return false;
      }
    
      SmallVector<Operation*> current_layer_descendants;
      SmallVector<Operation*> next_layer_descendants;
      int current_depth = 0;
      current_layer_descendants.push_back(op);
      // BFS downstream ops for current user.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 21K bytes
    - Viewed (0)
  10. 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)
Back to top