- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for _xla (0.03 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
bool ShapeInference::InferShapeForXlaGatherOp(XlaGatherOp op) { xla::Shape input_shape = xla::TypeToShape(op.getOperand().getType()); if (input_shape == xla::Shape() || input_shape.is_unbounded_dynamic()) return false; xla::Shape start_indices_shape = xla::TypeToShape(op.getStartIndices().getType()); if (start_indices_shape == xla::Shape()) return false; xla::GatherDimensionNumbers gather_dim_numbers;
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/transforms/tf_passes.td
} def TPUValidateInputsPass : Pass<"tf-tpu-validate-inputs", "ModuleOp"> { let summary = "Validates inputs to the TPU TF/XLA bridge"; let description = [{ This pass checks that the IR has valid input to TPU TF/XLA bridge. It checks the relations of multiple ops. Properties of single ops are checked by the 'verify' method of ops. }];
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 99.6K bytes - Viewed (0)