- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for ShapedTypeComponents (0.43 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc
static LogicalResult inferConvReturnTypeComponents( std::optional<mlir::Location> location, OpT op, ArrayRef<Attribute> explicit_padding, llvm::SmallVectorImpl<ShapedTypeComponents>& inferredReturnShapes) { const int64_t num_spatial_dims = std::is_same<OpT, Conv2DOpAdaptor>() ? 2 : 3; const int64_t num_dims = 2 + num_spatial_dims; const Value input = op.getInput();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 146.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
auto result_element_type_fn = [&](int index) { return mlir::cast<TensorType>(op->getResult(index).getType()) .getElementType(); }; llvm::SmallVector<ShapedTypeComponents, 4> inferred_return_shapes; if (failed(InferReturnTypeComponentsForTFOp( /*location=*/std::nullopt, op, graph_version_, operand_as_constant_fn, op_result_as_shape_fn, result_element_type_fn,
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/tensorflow/ir/tf_ops_n_z.cc
MLIRContext *context, std::optional<Location> location, ValueShapeRange operands, DictionaryAttr attributes, OpaqueProperties, RegionRange regions, SmallVectorImpl<ShapedTypeComponents> &inferredReturnShapes) { XlaBroadcastHelperOpAdaptor op(operands.getValues(), attributes); Value lhs = op.getLhs(); Value rhs = op.getRhs(); auto set_unranked_results = [&]() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0)