Searched refs:src1_tensor (Results 1 – 1 of 1) sorted by relevance
52 const Tensor& src1_tensor = MklGetInput(ctx, src1_idx); in Compute() local61 : src1_tensor.dims(); in Compute()67 : src1_tensor.shape(); in Compute()85 src1_tensor.shape(), mkl_shape_dst); in Compute()86 float user_i1 = (src1_tensor.scalar<T>()()); in Compute()94 if (src1_tensor.shape().num_elements() == 0) { in Compute()99 src1_tensor.shape(), mkl_shape_dst); in Compute()127 dims = TFShapeToMklDnnDims(src1_tensor.shape()); in Compute()156 if (src1_tensor.dims() == 4) { in Compute()157 src1_dims = TFShapeToMklDnnDimsInNCHW(src1_tensor.shape(), in Compute()[all …]