- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for output_dimensions (0.24 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_patterns.td
// Dynamism op def : Pat<(MHLO_DynamicBroadcastInDimOp:$output $input, $output_dimensions, $broadcast_dimensions, $expanding_dimensions_unused, $nonexpanding_dimensions_unused), (TF_BroadcastToOp $input, $output_dimensions), [(IsTFStyleBroadcast $broadcast_dimensions, $output)]>; def : Pat<(MHLO_DynamicBroadcastInDimOp:$output $input, $output_dimensions,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Feb 03 08:58:22 UTC 2024 - 34K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
const int64_t dilated_window = xla::window_util::DilatedBound(dim.size(), dim.window_dilation()); output_dimensions[i] = xla::window_util::StridedBound( padded_dilated_base, dilated_window, dim.stride()); } } return tensorflow::GetTypeFromTFTensorShape(output_dimensions, element_type); } bool ShapeInference::InferShapeForXlaReduceWindowOp(XlaReduceWindowOp op) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0)