Searched refs:src_tf_shape (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_matmul_op_fused.cc | 71 auto src_tf_shape = src_mkl_shape.IsMklTensor() ? src_mkl_shape.GetTfShape() in Compute() local
76 OP_REQUIRES(ctx, TensorShapeUtils::IsMatrix(src_tf_shape), in Compute()
89 const int batch = src_tf_shape.dim_size(1 - dim_pair[0]); in Compute()
90 const int k = src_tf_shape.dim_size(dim_pair[0]); in Compute()
96 "Matrix size-incompatible: In[0]: ", src_tf_shape.DebugString(), in Compute()
|
| D | mkl_conv_grad_input_ops.cc | 331 TensorShape src_tf_shape; in Compute() local
336 &src_tf_shape)); in Compute()
338 src_tf_shape = MakeInputTfShape(context, src_tensor); in Compute()
347 if (src_tf_shape.num_elements() == 0 || in Compute()
353 GetOutputTfShape(src_tf_shape, filter_tf_shape, diff_dst_tf_shape); in Compute()
378 src_tf_shape, filter_tf_shape, &fwd_src_dims, &fwd_filter_dims, in Compute()
|
| D | mkl_softmax_op.cc | 211 auto src_tf_shape = src_mkl_shape.IsMklTensor() in Compute() local
214 const int input_dims = src_tf_shape.dims(); in Compute()
221 src_dims = TFShapeToMklDnnDims(src_tf_shape); in Compute()
|
| D | mkl_conv_grad_filter_ops.cc | 388 TensorShape src_tf_shape = MakeInputTfShape(context, src_tensor); in Compute() local
395 if (src_tf_shape.num_elements() == 0 || in Compute()
401 GetOutputTfShape(src_tf_shape, filter_tf_shape, diff_dst_tf_shape); in Compute()
426 src_tf_shape, filter_tf_shape, &fwd_src_dims, &fwd_filter_dims, in Compute()
|
| D | mkl_qmatmul_op.cc | 181 auto src_tf_shape = src_mkl_shape.IsMklTensor() in Compute() local
188 src_dims = TFShapeToMklDnnDims(src_tf_shape); in Compute()
190 dst_dims_mkl_order = {static_cast<int>(src_tf_shape.dim_size(0)), in Compute()
|
| D | mkl_quantize_op.cc | 334 auto src_tf_shape = src_mkl_shape.IsMklTensor() ? src_mkl_shape.GetTfShape() in Compute() local
342 switch (src_tf_shape.dims()) { in Compute()
|
| D | mkl_conv_ops.cc | 537 auto src_tf_shape = GetTfShape(context, kInputIndex_Src, native_format); in Compute() local
541 src_tf_shape, filter_tf_shape, &src_dims, &filter_dims, &strides, in Compute()
561 src_tf_shape, dst_mkl_shape, native_format); in Compute()
|