Searched refs:idx_in (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 813 inline void CopyMklTensorInToOut(OpKernelContext* context, int idx_in, in CopyMklTensorInToOut() argument
817 int idx_data_in = GetTensorDataIndex(idx_in, num_inputs); in CopyMklTensorInToOut()
818 int idx_meta_in = GetTensorMetaDataIndex(idx_in, num_inputs); in CopyMklTensorInToOut()
834 inline void CopyTfTensorInToOutWithShape(OpKernelContext* context, int idx_in, in CopyTfTensorInToOutWithShape() argument
839 int idx_data_in = GetTensorDataIndex(idx_in, num_inputs); in CopyTfTensorInToOutWithShape()
852 inline void ForwardTfTensorInToOut(OpKernelContext* context, int idx_in, in ForwardTfTensorInToOut() argument
856 int idx_data_in = GetTensorDataIndex(idx_in, num_inputs); in ForwardTfTensorInToOut()
869 inline void ForwardMklTensorInToOut(OpKernelContext* context, int idx_in, in ForwardMklTensorInToOut() argument
873 int idx_data_in = GetTensorDataIndex(idx_in, num_inputs); in ForwardMklTensorInToOut()
874 int idx_meta_in = GetTensorMetaDataIndex(idx_in, num_inputs); in ForwardMklTensorInToOut()
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | special_math_ops.py | 1080 idx_in, idx_out = _einsum_v1_parse_and_resolve_equation(
1083 idx_all = set(''.join(idx_in) + idx_out)
1098 for i, (input_, axes_) in enumerate(zip(inputs, idx_in)):
1116 idx_in[i] = sorted_idx
1126 for i, idx in enumerate(idx_in):
|