Search Options

Results per page
Sort
Preferred Languages
Advance

Results 171 - 180 of 345 for SmallVector (0.32 sec)

  1. tensorflow/compiler/mlir/lite/transforms/optimize_batch_matmul.cc

    limitations under the License.
    ==============================================================================*/
    
    #include <algorithm>
    #include <cstdint>
    #include <memory>
    #include <utility>
    
    #include "llvm/ADT/SmallVector.h"
    #include "llvm/Support/Casting.h"
    #include "mlir/Dialect/Arith/IR/Arith.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: Thu Apr 25 16:01:03 UTC 2024
    - 9.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/ir/tf_op_interfaces.h

      // Indices of additional arguments that will be forwarded to the fused
      // operation (e.g. forward bias vector if fusing BiasAdd operation).
      SmallVector<int, 4> additional_arguments;
    
      // Add additional attributes to the fused node.
      SmallVector<NamedAttribute, 4> additional_attributes;
    };
    
    //===----------------------------------------------------------------------===//
    // TensorFlow Resource Handles.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 03 19:26:14 UTC 2023
    - 6.5K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting.cc

      int num_old_results = op.getNumResults();
      int num_new_results = hoister.GetLiftedNumResults();
      int num_extra_results = num_new_results - num_old_results;
    
      SmallVector<Type, 4> new_result_types;
      SmallVector<Value, 4> new_while_operands;
      new_result_types.resize(num_extra_results);
      new_while_operands.resize(num_extra_results);
    
      for (auto& it : hoister.GetResources()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 55.1K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/transforms/prepare_quantize_helper.h

        // the reduce operation after element-wise multiplication between state and
        // time weights.
        if (tensor_property.number_of_bits == 10) {
          SmallVector<double, 4> mins(1, std::numeric_limits<double>::max());
          SmallVector<double, 4> maxs(1, std::numeric_limits<double>::min());
          // Computes the effective min/max values of the attribute values.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 03 18:01:23 UTC 2024
    - 28K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/utils/xla_rewrite_util_test.cc

      llvm::SmallVector<mlir::tf_device::ClusterFuncOp, 4> cluster_func_ops;
      module->walk([&](mlir::tf_device::ClusterFuncOp cluster_func) {
        cluster_func_ops.push_back(cluster_func);
      });
      EXPECT_EQ(cluster_func_ops.size(), 1);
    
      EXPECT_TRUE(mlir::succeeded(tensorflow::EraseClusterFuncs(cluster_func_ops)));
    
      llvm::SmallVector<mlir::tf_device::ClusterFuncOp, 4> new_cluster_func_ops;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Apr 26 09:37:10 UTC 2024
    - 4.3K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/transforms/modify_io_nodes.cc

        return;
      }
    
      OpBuilder builder(func);
      FunctionType func_type = func.getFunctionType();
      llvm::SmallVector<Type, 4> new_input_types(func_type.getInputs().begin(),
                                                 func_type.getInputs().end());
      llvm::SmallVector<Type, 4> new_output_types(func_type.getResults().begin(),
                                                  func_type.getResults().end());
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.9K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/lite/quantization/import_quant_stats_pass.cc

    limitations under the License.
    ==============================================================================*/
    
    #include <memory>
    
    #include "absl/strings/str_split.h"
    #include "llvm/ADT/APFloat.h"
    #include "llvm/ADT/SmallVector.h"
    #include "llvm/ADT/StringMap.h"
    #include "llvm/Support/CommandLine.h"
    #include "llvm/Support/Regex.h"
    #include "mlir/Dialect/Func/IR/FuncOps.h"  // from @llvm-project
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Mar 08 10:41:08 UTC 2024
    - 9.9K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/lite/stablehlo/transforms/unfuse_batch_norm_pass.cc

    limitations under the License.
    ==============================================================================*/
    
    #include <cstdint>
    #include <memory>
    #include <utility>
    
    #include "llvm/ADT/SmallVector.h"
    #include "mlir/Dialect/Func/IR/FuncOps.h"  // from @llvm-project
    #include "mlir/Dialect/Shape/IR/Shape.h"  // from @llvm-project
    #include "mlir/IR/Attributes.h"  // from @llvm-project
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 11.2K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/transforms/xla_call_module_deserialization.cc

    #include <memory>
    #include <string>
    #include <utility>
    #include <vector>
    
    #include "llvm/ADT/ArrayRef.h"
    #include "llvm/ADT/STLExtras.h"
    #include "llvm/ADT/SmallVector.h"
    #include "llvm/ADT/StringRef.h"
    #include "mlir/Dialect/Func/Extensions/AllExtensions.h"  // from @llvm-project
    #include "mlir/Dialect/Func/IR/FuncOps.h"  // from @llvm-project
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 23 09:05:47 UTC 2024
    - 11.1K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/transforms/constant_fold.cc

            return !attr || !mlir::isa<ElementsAttr>(attr);
          }))
        return failure();
    
      SmallVector<ElementsAttr, 4> inputs;
      inputs.reserve(operands.size());
      for (auto input : operands) {
        inputs.push_back(mlir::cast<ElementsAttr>(input));
      }
    
      SmallVector<Attribute> constants;
      LogicalResult status = EvaluateOperation(inst, inputs, constants);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 5.3K bytes
    - Viewed (0)
Back to top