Home
last modified time | relevance | path

Searched defs:tensor_shape0 (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_input_conversion_op.cc88 TensorShape tensor_shape0 = input_tensor_0.shape(); in Compute() local
/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/
H A DCpuGemmMatrixMultiplyKernel.cpp93 TensorShape tensor_shape0{ lhs->tensor_shape() }; in validate_arguments() local
/aosp_15_r20/external/ComputeLibrary/src/gpu/cl/kernels/
H A DClGemmLowpMatrixMultiplyReshapedKernel.cpp75 TensorShape tensor_shape0{ src0->tensor_shape() }; in validate_arguments() local
H A DClGemmMatrixMultiplyReshapedKernel.cpp103 TensorShape tensor_shape0{ src0->tensor_shape() }; in validate_arguments() local