Searched defs:inner_dim_size (Results 1 – 3 of 3) sorted by relevance
175 const int64_t input_outer_dim_size, const int64_t inner_dim_size, in UnsortedSegmentCustomKernel()
297 int32 count, int32 channels_size, int32 inner_dim_size, const T* in, in FusedBatchNormInferenceMetaKernel()
381 int64_t inner_dim_size = EIGEN_MAX_ALIGN_BYTES + 1; in TEST_F() local