Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for output_dimensions (0.24 sec)

  1. 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)
  2. 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)
Back to top