- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for in_shape (0.09 sec)
-
tensorflow/compiler/mlir/tf2xla/internal/passes/extract_outside_compilation.cc
ArrayRef<int64_t> in_shape = ranked_type.getShape(); if (in_shape.empty() || in_shape[0] < 0) { return context_op->emitOpError() << "A map_outside_compilation op's input and output shapes must " "have rank at least one and the first dimension must be known."; } int64_t split_size = in_shape[0] / num_cores_per_replica; if (in_shape[0] % num_cores_per_replica != 0) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 68.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/python/integration_test/quantize_model_test.py
) def test_einsum_ptq_model( self, equation: str, ): _, y_shape, bias_shape, x_signature, y_signature = ( self._prepare_sample_einsum_datashapes(equation, use_bias=True) ) model = self._create_einsum_model( self._input_saved_model_path, equation, y_shape, x_signature, y_signature, bias_shape, )
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 06:31:57 UTC 2024 - 51.4K bytes - Viewed (0)