Searched defs:shape_gemm (Results 1 – 2 of 2) sorted by relevance
252 TensorShape shape_gemm; in configure() local463 TensorShape shape_gemm; in validate() local
336 TensorShape shape_gemm; in configure() local524 TensorShape shape_gemm = gemm_input_to_use->tensor_shape(); in validate() local