- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 147 for SmallVector (0.17 sec)
-
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) -
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) -
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) -
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) -
tensorflow/compiler/mlir/tf2xla/internal/passes/extract_head_tail_outside_compilation.cc
void RemoveClusterAliasedOutputs(OpBuilder* builder, mlir::tf_device::ClusterOp cluster) { llvm::SmallVector<Value, 4> used_old_cluster_results; llvm::SmallVector<Value, 4> new_cluster_results; llvm::SmallVector<Type, 4> new_cluster_result_types; Operation* cluster_terminator = cluster.GetBody().getTerminator(); for (auto result :
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 20.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/layout_optimization.cc
// Move Transpose operations that permute `op` operands after the `op`. void MoveTransposeAfter(Operation* op, SmallVector<Operation*, 8>* work_list, bool fold_transpose_in_ops) { // Indices of operands and results that depend on data layout. SmallVector<unsigned, 4> layout_dependent_operands; SmallVector<unsigned, 4> layout_dependent_results; auto fold_operands = dyn_cast<FoldOperandsTransposeInterface>(op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 19.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_cluster_formation.cc
mlir::tf_device::ReplicateOp replicate_op, PartitionedClusterOutputMap& partitioned_outputs, llvm::SmallVector<mlir::TF::IdentityOp, 8>& erase_list, llvm::SmallVector<Type, 8>& result_types, int num_replicas) { Operation* result_op; llvm::SmallVector<Value, 8> results; uint64_t num_results = cluster.getNumResults(); for (uint64_t result_id = 0; result_id < num_results; ++result_id) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 39.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/device_transform_patterns.cc
// Pack op should have same shape type. SmallVector<Value, 5> pack_inputs(pack_op.getValues()); auto input_type = mlir::dyn_cast<RankedTensorType>(pack_inputs[0].getType()); if (!input_type) return failure(); // Figure out output shapes. SmallVector<int64_t, 4> concat_out_shape; SmallVector<int64_t, 4> pack_out_shape; const int64_t rank = input_type.getRank();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 25.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/call_graph_util_test.cc
ASSERT_TRUE(module); mlir::SymbolTable symtab(*module); llvm::SmallVector<mlir::func::FuncOp> entry_funcs = GetEntryFunctions(*module); EXPECT_EQ(entry_funcs.size(), 1); EXPECT_EQ(entry_funcs[0].getSymName(), "entry_func"); llvm::SmallVector<mlir::SymbolUserOpInterface> outermost_pcall_ops; auto result = mlir::GetFirstOpsOfType<mlir::TF::StatefulPartitionedCallOp,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Dec 20 04:39:18 UTC 2023 - 11.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/while_loop_outline.cc
const int operands_size = while_op.getNumOperands() + extra_operands.size(); SmallVector<Value, 4> operands; operands.reserve(operands_size); operands.append(while_op.getOperands().begin(), while_op.getOperands().end()); operands.append(extra_operands.begin(), extra_operands.end()); SmallVector<Type, 4> new_types; new_types.reserve(operands_size); new_types.append(while_op.getResultTypes().begin(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.8K bytes - Viewed (0)