- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for SmallVector (0.54 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
} std::optional<RankedTensorType> InferXlaConvOutputShape( llvm::SmallVector<int64_t> input_tensor_dims, llvm::SmallVector<int64_t> kernel_tensor_dims, llvm::SmallVector<int64_t> window_strides, llvm::SmallVector<std::pair<int64_t, int64_t>> paddings, llvm::SmallVector<int64_t> lhs_dilations, llvm::SmallVector<int64_t> rhs_dilations, int64_t batch_group_count,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_export.cc
auto dict_attr = fn->getAttrOfType<mlir::DictionaryAttr>("tf.entry_function"); if (!dict_attr) return; llvm::SmallVector<llvm::StringRef, 2> input_names; llvm::SmallVector<llvm::StringRef, 2> output_names; if (auto str = mlir::dyn_cast_or_null<mlir::StringAttr>(dict_attr.get("inputs"))) { str.getValue().split(input_names, ',', /*MaxSplit=*/-1,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (0)