| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_softmax_op.cc | 41 memory::dims src_dims; member in tensorflow::MklSoftmaxParams
47 MklSoftmaxParams(memory::dims src_dims, MklTensorFormat src_fmt, int axis) in MklSoftmaxParams()
229 memory::dims src_dims; in Compute() local
|
| D | mkl_eltwise_activation_base_op.h | 61 MklEltwiseFwdActivationParams(memory::dims src_dims, memory::desc src_md, in MklEltwiseFwdActivationParams()
267 memory::dims src_dims; in Compute() local
|
| D | mkl_relu_op.cc | 47 memory::dims src_dims; member in tensorflow::MklEltwiseFwdParams
53 MklEltwiseFwdParams(memory::dims src_dims, memory::desc src_md, in MklEltwiseFwdParams()
223 memory::dims src_dims; member in tensorflow::MklEltwiseBwdParams
232 MklEltwiseBwdParams(const memory::dims& src_dims, in MklEltwiseBwdParams()
464 memory::dims src_dims; in Compute() local
622 memory::dims src_dims = {}; in Compute() local
|
| D | mkl_dequantize_op.cc | 67 auto src_dims = TFShapeToMklDnnDims(src_tensor.shape()); in Compute() local
|
| D | mkl_concat_op.cc | 243 std::vector<memory::dims> src_dims; member
638 memory::dims src_dims(src_md.data.dims, in Compute() local
650 memory::dims src_dims = TFShapeToMklDnnDims(s_shape); in Compute() local
897 auto src_dims = TFShapeToMklDnnDims(input_shapes[k].GetTfShape()); in FindMklCommonFormat() local
|
| D | mkl_layer_norm_op.cc | 66 memory::dims src_dims = TFShapeToMklDnnDims(src_tensor.shape()); in Compute() local
|
| D | mkl_quantize_op.cc | 62 memory::dims src_dims; member
345 auto src_dims = src_mkl_shape.IsMklTensor() in Compute() local
|
| D | mkl_matmul_op_fused.cc | 116 memory::dims src_dims = memory::dims({batch, k}); in Compute() local
|
| D | mkl_aggregate_ops.cc | 194 memory::dims src_dims; in Compute() local
|
| D | mkl_fused_batch_norm_op.cc | 48 memory::dims src_dims; member
392 memory::dims src_dims; member
828 memory::dims src_dims = in Compute() local
1252 memory::dims src_dims = in Compute() local
|
| D | mkl_avgpooling_op.cc | 88 memory::dims src_dims = in Compute() local
|
| D | mkl_maxpooling_op.cc | 114 memory::dims src_dims = in Compute() local
|
| D | mkl_matmul_ops_common.h | 60 memory::dims src_dims; member
85 : src_dims(src_dims), in src_dims() function
|
| D | mkl_qmatmul_op.cc | 178 memory::dims src_dims, weight_dims; in Compute() local
|
| D | mkl_conv_grad_filter_ops.cc | 43 memory::dims src_dims; member
|
| D | mkl_pooling_ops_common.h | 43 memory::dims src_dims; member
|
| D | mkl_conv_ops.cc | 43 memory::dims src_dims; member
648 memory::dims src_dims, filter_dims, padding_left, padding_right, in Compute() local
|