- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for IsXlaGatherWithoutBatch (0.3 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/convert_tf_xla_op_to_tf_op.cc
return RestoreCollapsedDimensions(ranked_output_type, collapsed_dims); } return xla_gather_op_output_type; } // TODO (b/275225582): Supports Xla Gather op in general case. bool IsXlaGatherWithoutBatch(Value operand, Value start_indices) { auto operand_type = mlir::dyn_cast_or_null<ShapedType>(operand.getType()); auto start_indices_type = mlir::dyn_cast_or_null<ShapedType>(start_indices.getType());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 13.2K bytes - Viewed (0)