Home
last modified time | relevance | path

Searched defs:tmp_shape (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dlist_kernels.h801 TensorShape tmp_shape = tmp.shape(); in Compute() local
825 TensorShape tmp_shape = tmp.shape(); in Scatter() local
/aosp_15_r20/external/ComputeLibrary/src/gpu/cl/kernels/
H A DClGemmReshapeLhsMatrixKernel.cpp85 TensorShape tmp_shape(src->tensor_shape()); in configure_window() local
H A DClGemmLowpMatrixMultiplyReshapedKernel.cpp118 TensorShape tmp_shape(dst->tensor_shape()); in validate_and_configure_window() local
H A DClGemmLowpMatrixMultiplyNativeKernel.cpp132 TensorShape tmp_shape(dst->tensor_shape()); in validate_and_configure_window() local
H A DClGemmMatrixMultiplyReshapedOnlyRhsMMULKernel.cpp162 TensorShape tmp_shape(dst->tensor_shape()); in validate_and_configure_window() local
H A DClGemmMatrixMultiplyReshapedKernel.cpp162 TensorShape tmp_shape(dst->tensor_shape()); in validate_and_configure_window() local
H A DClGemmMatrixMultiplyReshapedOnlyRhsKernel.cpp162 TensorShape tmp_shape(dst->tensor_shape()); in validate_and_configure_window() local
H A DClGemmMatrixMultiplyNativeKernel.cpp170 TensorShape tmp_shape(dst->tensor_shape()); in validate_and_configure_window() local
H A DClGemmLowpMatrixMultiplyReshapedOnlyRhsMMULKernel.cpp216 TensorShape tmp_shape(dst->tensor_shape()); in validate_and_configure_window() local
H A DClGemmLowpMatrixMultiplyReshapedOnlyRhsKernel.cpp236 TensorShape tmp_shape(dst->tensor_shape()); in validate_and_configure_window() local