- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 66 for SmallVectorImpl (0.27 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/tf_xla_mlir_translate.cc
if (resource_update.modified) output << " modified"; output << '\n'; } return mlir::success(); } Status ParseArgumentShapes( absl::string_view input_shapes_str, llvm::SmallVectorImpl<TensorOrResourceShape>& arg_shapes) { arg_shapes.clear(); std::vector<std::optional<std::vector<int>>> input_shapes_vector; TF_RETURN_IF_ERROR(ParseNodeShapes(input_shapes_str, input_shapes_vector));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 18.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/shape_inference_utils.cc
tfg::OperandAsConstantFn operand_as_constant_fn, tfg::OpResultAsShapeFn op_result_as_shape_fn, tfg::ResultElementTypeFn result_element_type_fn, SmallVectorImpl<ShapedTypeComponents>& inferred_return_shapes) { assert(op->getName().getDialectNamespace() == TensorFlowDialect::getDialectNamespace()); return tfg::InferReturnTypeComponentsForTFOp(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Dec 22 13:13:44 UTC 2022 - 1.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/fold_broadcast_pass.cc
// 'shape' is computed from the original shape and the broadcast dimensions to // match result shape. int64_t GetElementIndex(llvm::SmallVectorImpl<int64_t> &shape, llvm::SmallVectorImpl<int64_t> ¤t_index) { int64_t ind = 0; int64_t mul = 1; for (int i = shape.size() - 1; i >= 0; --i) { ind += (current_index[i] % shape[i]) * mul; mul *= shape[i];
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/eval_util.h
// results vector. If not, results vector is unspecified. // mlir::LogicalResult EvaluateOperation( mlir::Operation* inst, llvm::ArrayRef<mlir::ElementsAttr> operands, TFE_Context* context, llvm::SmallVectorImpl<mlir::Attribute>* results); } // namespace tensorflow
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Dec 19 06:31:40 UTC 2020 - 1.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_operator.h
mlir::Builder builder, // NOLINTNEXTLINE Location loc, llvm::SmallVectorImpl<mlir::NamedAttribute> *attributes); // TODO(zichuanwei@): Populate Builtin_options_2 manual for now, should automate // these in the future void BuiltinOptions2ToAttributes( tflite::BuiltinOptions2Union op_union, mlir::Builder builder, llvm::SmallVectorImpl<mlir::NamedAttribute> &attributes);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 21:00:09 UTC 2024 - 11.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/shape_inference_utils.h
tfg::OperandAsConstantFn operand_as_constant_fn, tfg::OpResultAsShapeFn op_result_as_shape_fn, tfg::ResultElementTypeFn result_element_type_fn, SmallVectorImpl<ShapedTypeComponents>& inferred_return_shapes); } // namespace TF } // namespace mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 06 04:50:13 UTC 2023 - 1.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/convert_type.cc
llvm::SmallVectorImpl<int64_t>* shape) { shape->reserve(input_shape.dims()); for (const auto& d : input_shape) { shape->push_back(d.size == kTFDynamicSize ? ShapedType::kDynamic : d.size); } } Status ConvertToMlirShape(const TensorShapeProto& input_shape, llvm::SmallVectorImpl<int64_t>* shape) { shape->reserve(input_shape.dim_size());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 7.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
// the sliced type from StridedSlice. return success(); } bool StridedSliceGradOp::GetSlicedShapeAndBoundRanges( SmallVectorImpl<int64_t> *input_shape, SmallVectorImpl<int64_t> *slice_begin, SmallVectorImpl<int64_t> *slice_end, SmallVectorImpl<int64_t> *slice_stride) { DenseIntElementsAttr shape_attr; DenseIntElementsAttr sparse_begin_attr, sparse_end_attr, sparse_strides_attr;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_utils.cc
// range. void ExpandVerySmallRange(const ArrayRef<double> mins, const ArrayRef<double> maxs, SmallVectorImpl<double>& effective_mins, SmallVectorImpl<double>& effective_maxs) { for (const auto [min, max] : llvm::zip(mins, maxs)) { // The range is small. Expands the range to stride 0.0 and also at least // 1.0e-6.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 02:10:16 UTC 2024 - 43.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/remove_vars_in_session_initializer.cc
RemoveVariablesInSessionInitializerPass> { public: void runOnOperation() override; }; void RecursiveRemove(Operation* op, llvm::SmallVectorImpl<Operation*>& erase_list, llvm::SmallPtrSetImpl<Operation*>& dead_ops) { for (mlir::Value res : op->getResults()) { for (Operation* user : res.getUsers()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.6K bytes - Viewed (0)