- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 171 for ArrayRef (0.19 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/collection_ops_util.cc
if (element.getType() != slice_type) { update_slice = builder.create<TF::ReshapeOp>( loc, ArrayRef<Type>{slice_type}, ArrayRef<Value>{element, GetR1Const(slice_shape, builder, loc)}); } return builder .create<TF::XlaDynamicUpdateSliceOp>( loc, ArrayRef<Type>{buffer.getType()}, ArrayRef<Value>{buffer, update_slice, GetIndicesForElement(index, buffer, builder, loc)})
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference_pass.cc
<< max_iterations_; return signalPassFailure(); } } private: ArrayRef<ArrayRef<int64_t>> input_shapes_; }; } // namespace std::unique_ptr<OperationPass<ModuleOp>> CreateTFShapeInferencePass( ArrayRef<ArrayRef<int64_t>> input_shapes) { return std::make_unique<ShapeInference>(input_shapes); } } // namespace TF
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 12:49:45 UTC 2024 - 3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.h
// InferShapeForFunction. FailureOr<bool> InferModuleShape(ModuleOp module, int64_t max_iterations = 10, ArrayRef<TypeID> ops_to_skip = {}, ArrayRef<ArrayRef<int64_t>> input_shapes = {}); // Given a tensorflow NodeShape string, returns a vector of argument shapes // that can be used with InferShapeForFunction.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 12:49:45 UTC 2024 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/extract_outside_compilation.cc
// creating comm ops. LogicalResult MoveToHostSingleCluster( ArrayRef<Operation*> clustered_ops, ArrayRef<Value> external_operands, ArrayRef<Value> external_outputs, llvm::SmallVector<IRMapping>& core_to_mapping, ArrayRef<Operation*> core_to_host_insertion_point, ArrayRef<Value> core_to_compilation_key, ArrayRef<Value> core_to_device_ordinal, StringAttr device_type_attr,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 68.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/quantize_preprocess.h
mlir::ModuleOp module_op, mlir::MLIRContext* context, std::optional<Session*> session, bool run_tf_to_stablehlo, bool deserialize_xla_call_module, llvm::ArrayRef<llvm::ArrayRef<int64_t>> input_arg_shapes = {}); // Overload of `PreprocessAndFreezeGraph` that uses the default MLIR dump file // prefix. inline absl::Status PreprocessAndFreezeGraph(mlir::ModuleOp module_op,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 12:49:45 UTC 2024 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/fold_broadcast.cc
template <typename Op> LogicalResult RewriteEqOp(Operation* op, PatternRewriter& rewriter) const; LogicalResult RewriteOp( Operation* op, PatternRewriter& rewriter, const std::function<bool(ArrayRef<int64_t>, ArrayRef<int64_t>, SmallVectorImpl<int64_t>&)>& get_broadcasted_shape) const; LogicalResult RewriteBatchMatMulV2Op(Operation* op,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/utils.h
namespace mlir { namespace TFL { using llvm::ArrayRef; using mlir::Operation; using mlir::ShapedType; using mlir::Value; // Returns true if all tensor value in `values` has static shape and same shape. inline bool OpHasSameStaticShapes(Operation* op) { auto values = op->getOperands(); int operand_num = 0; ArrayRef<int64_t> shape; for (Value value : values) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 11.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/optimize_layout.cc
static SmallVector<int64_t> InvertPermutationToVector(ArrayRef<int64_t> vec, ArrayRef<int64_t> perm) { return applyPermutation(vec, invertPermutationVector(perm)); } static RankedTensorType GetPermutedTensorTypeHelper(RankedTensorType type, ArrayRef<int64_t> perm,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 21:59:06 UTC 2024 - 8.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/readonly_references_to_resources.cc
} VarHandleOp var_handle_op = builder.create<VarHandleOp>( variable_v2_op.getLoc(), ArrayRef<Type>{RankedTensorType::get( {}, TF::ResourceType::get(ArrayRef<TensorType>{tensor_type}, builder.getContext()))}, ArrayRef<Value>{}, ArrayRef<NamedAttribute>{ builder.getNamedAttr("device", device_attr), builder.getNamedAttr("container",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_layout_helper.cc
return ArrayAttr::get(attr.getContext(), shuffled); } // Shuffle ranked tensor dimensions according to the permutation. Type ShuffleRankedTensorType(Type type, ArrayRef<int64_t> permutation) { if (auto ranked_type = mlir::dyn_cast<RankedTensorType>(type)) { ArrayRef<int64_t> shape = ranked_type.getShape(); assert(permutation.size() == shape.size()); SmallVector<int64_t, 4> new_shape(permutation.size());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.3K bytes - Viewed (0)