Searched defs:weight_shape (Results 1 – 5 of 5) sorted by relevance
38 std::initializer_list<int> weight_shape, in BaseEmbeddingLookupOpModel()81 std::initializer_list<int> weight_shape, in HybridEmbeddingLookupOpModel()
34 std::initializer_list<int> weight_shape) { in LSHProjectionOpModel()
40 const TensorShape& weight_shape, in BuildGraphDef()
372 auto weight_shape = in TestFoldFusedBatchNormsWithConcat() local
1201 std::initializer_list<int> weight_shape) { in LSHProjectionOpModel()3462 std::initializer_list<int> weight_shape, in BaseEmbeddingLookupOpModel()