Home
last modified time | relevance | path

Searched defs:tensor_shape1 (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_input_conversion_op.cc89 TensorShape tensor_shape1 = input_tensor_1.shape(); in Compute() local
/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/
H A DCpuGemmMatrixMultiplyKernel.cpp103 TensorShape tensor_shape1{ rhs->tensor_shape() }; in validate_arguments() local
/aosp_15_r20/external/ComputeLibrary/src/gpu/cl/kernels/
H A DClGemmLowpMatrixMultiplyReshapedKernel.cpp79 TensorShape tensor_shape1{ src1->tensor_shape() }; in validate_arguments() local
H A DClGemmMatrixMultiplyReshapedOnlyRhsMMULKernel.cpp127 TensorShape tensor_shape1{ src1->tensor_shape() }; in validate_arguments() local
H A DClGemmMatrixMultiplyReshapedKernel.cpp107 TensorShape tensor_shape1{ src1->tensor_shape() }; in validate_arguments() local
H A DClGemmMatrixMultiplyReshapedOnlyRhsKernel.cpp93 TensorShape tensor_shape1{ src1->tensor_shape() }; in validate_arguments() local
H A DClGemmLowpMatrixMultiplyReshapedOnlyRhsMMULKernel.cpp72 TensorShape tensor_shape1{ src1->tensor_shape() }; in validate_arguments() local
H A DClGemmLowpMatrixMultiplyReshapedOnlyRhsKernel.cpp87 TensorShape tensor_shape1{ src1->tensor_shape() }; in validate_arguments() local