- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for BroadcastDimsProductEqual (0.38 sec)
-
tensorflow/compiler/mlir/lite/transforms/optimize_patterns.td
]>; // Return true if the product of dimension values of a subsection of the tensor // is equal to the non-contracting dimension after a reshape class BroadcastDimsProductEqual<int agg_start_idx> : Constraint<CPred< "TFL::BroadcastDimsProductEqual($0, $1, "# agg_start_idx #")">>; // Returns true if the dimensions of a subsection of two tensors is equal
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 20:31:41 UTC 2024 - 66.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize.cc
} void runOnOperation() override; }; // Return true if the product of dimension values of a subsection of the tensor // is equal to the non-contracting dimension after a reshape bool BroadcastDimsProductEqual(Value input, Value output, size_t agg_start_idx) { ArrayRef<int64_t> input_shape = mlir::cast<ShapedType>(input.getType()).getShape(); ArrayRef<int64_t> output_shape =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 102.3K bytes - Viewed (0)