Searched defs:src_dims (Results 1 – 15 of 15) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_softmax_op.cc | 38 memory::dims src_dims; member in tensorflow::MklSoftmaxParams
42 MklSoftmaxParams(memory::dims src_dims, MklTensorFormat src_fmt, int axis) in MklSoftmaxParams()
215 memory::dims src_dims; in Compute() local
|
| D | mkl_dequantize_op.cc | 68 auto src_dims = src_mkl_shape.IsMklTensor() in Compute() local
|
| D | mkl_relu_op.cc | 44 memory::dims src_dims; member in tensorflow::MklEltwiseFwdParams
50 MklEltwiseFwdParams(memory::dims src_dims, memory::desc src_md, in MklEltwiseFwdParams()
219 memory::dims src_dims; member in tensorflow::MklEltwiseBwdParams
228 MklEltwiseBwdParams(const memory::dims& src_dims, in MklEltwiseBwdParams()
463 memory::dims src_dims; in Compute() local
620 memory::dims src_dims = {}; in Compute() local
|
| D | mkl_concat_op.cc | 239 std::vector<memory::dims> src_dims; member
626 memory::dims src_dims(src_md.data.dims, in Compute() local
638 memory::dims src_dims = TFShapeToMklDnnDims(s_shape); in Compute() local
880 auto src_dims = TFShapeToMklDnnDims(input_shapes[k].GetTfShape()); in FindMklCommonFormat() local
|
| D | mkl_quantize_op.cc | 60 memory::dims src_dims; member
336 auto src_dims = src_mkl_shape.IsMklTensor() in Compute() local
|
| D | mkl_matmul_op_fused.cc | 107 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_avgpooling_op.cc | 90 memory::dims src_dims = in Compute() local
|
| D | mkl_fused_batch_norm_op.cc | 44 memory::dims src_dims; member
378 memory::dims src_dims; member
767 memory::dims src_dims = in Compute() local
1161 memory::dims src_dims = in Compute() local
|
| D | mkl_matmul_ops_common.h | 47 memory::dims src_dims; member
66 : src_dims(src_dims), in src_dims() function
|
| D | mkl_maxpooling_op.cc | 119 memory::dims src_dims = in Compute() local
|
| D | mkl_qmatmul_op.cc | 177 memory::dims src_dims, weight_dims; in Compute() local
|
| D | mkl_conv_grad_filter_ops.cc | 56 memory::dims src_dims; member
|
| D | mkl_conv_ops.cc | 59 memory::dims src_dims; member
514 memory::dims src_dims, filter_dims, padding_left, padding_right, in Compute() local
|
| D | mkl_pooling_ops_common.h | 41 memory::dims src_dims; member
|