Home
last modified time | relevance | path

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

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