Searched defs:idx_in (Results 1 – 4 of 4) sorted by relevance
820 inline void CopyMklTensorInToOut(OpKernelContext* context, int idx_in, in CopyMklTensorInToOut()841 inline void CopyTfTensorInToOutWithShape(OpKernelContext* context, int idx_in, in CopyTfTensorInToOutWithShape()859 inline void ForwardTfTensorInToOut(OpKernelContext* context, int idx_in, in ForwardTfTensorInToOut()876 inline void ForwardMklTensorInToOut(OpKernelContext* context, int idx_in, in ForwardMklTensorInToOut()
1221 for (int32_t idx_in = 0; idx_in < inner_size; ++idx_in) { in OpInnerBroadcast() local
1220 for (int32_t idx_in = 0; idx_in < inner_size; ++idx_in) { in OpInnerBroadcast() local