- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for expected_outputs (0.14 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/quantize_model_test.py
self.assertCountEqual( converted_model.signatures._signatures.keys(), {signature_key} ) expected_outputs = model.matmul(**model_inputs) got_outputs = converted_model.signatures[signature_key](**model_inputs) self.assertAllClose(expected_outputs, got_outputs, atol=1e-1) output_loader = saved_model_loader.SavedModelLoader( self._output_saved_model_path )
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 03:36:50 UTC 2024 - 235.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize.cc
} // Returns whether the resultant type of any broadcastable operation with // operands `a` and `b` matches `expected_output`. Returns false if `a` is not // broadcast-compatible with `b`. bool OperandsBroadcastToOutputType(Type a, Type b, Type expected_output) { Type output_element_type = mlir::cast<ShapedType>(expected_output).getElementType(); Type broadcasted_type = OpTrait::util::getBroadcastedType(a, b, output_element_type);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 102.3K bytes - Viewed (0)