Commit 52718ef7 authored by Pavithra Vijay's avatar Pavithra Vijay Committed by TensorFlower Gardener
Browse files

Add `axis` property to CosineProximity loss and metric. 

Refactor cosine proximity tests for better clarity.

PiperOrigin-RevId: 228564762
parent 9a5dcb4c
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment