| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_softmax_op.cc | 102 TensorShape output_tf_shape; // shape of output TF tensor. variable
114 output_tf_shape.AddDim((dst_pd.get_size() / sizeof(T)));
117 output_tf_shape = MklDnnDimsToTFShape(output_dims);
120 AllocateOutputSetMklShape(context, 0, &output_tensor, output_tf_shape,
|
| mkl_pooling_ops_common.h | 227 TensorShape output_tf_shape; local
230 output_tf_shape.AddDim(this->GetNumTElements(dst_pd));
232 output_tf_shape, output_mkl_shape);
295 TensorShape output_tf_shape; local
296 output_tf_shape.AddDim(this->GetNumTElements(dst_pd));
298 output_tf_shape, output_mkl_shape);
|
| mkl_aggregate_ops.cc | 458 TensorShape output_tf_shape; variable
474 output_tf_shape.AddDim((output_pd.get_size() / sizeof(T)));
477 output_tf_shape = src1_tensor.shape();
479 AllocateOutputSetMklShape(ctx, output_idx, &dst_tensor, output_tf_shape,
|
| mkl_conv_ops.cc | 545 TensorShape output_tf_shape = MklDnnDimsToTFShape(output_dims_tf_order);
549 if (output_tf_shape.num_elements() == 0 || output_dims_tf_order[0] == 0) {
682 TensorShape output_tf_shape; local
683 output_tf_shape.AddDim((dst_pd.get_size() / sizeof(T)));
686 output_tf_shape, output_mkl_shape);
[all...] |
| mkl_conv_grad_input_ops.cc | 472 TensorShape output_tf_shape; local
473 output_tf_shape.AddDim(dst_pd.get_size() / sizeof(T));
475 AllocateOutputSetMklShape(context, 0, output_tensor, output_tf_shape,
|
| mkl_conv_ops.h | 391 TensorShape output_tf_shape = GetOutputTfShape( variable
395 output_tf_shape, output_mkl_shape);
399 for (size_t i = 0; i < output_tf_shape.num_elements(); ++i) {
|
| mkl_avgpooling_op.cc | 472 TensorShape output_tf_shape; variable
474 output_tf_shape = MklDnnDimsToTFShape(output_dims_mkl_order);
481 output_tf_shape = MklDnnDimsToTFShape(output_dims_NHWC_order);
485 output_tf_shape, output_mkl_shape);
|
| mkl_lrn_op.cc | [all...] |
| mkl_conv_grad_filter_ops.cc | 583 TensorShape output_tf_shape({output_dims_mkl_order[MklDnnDims::Dim_H],
587 AllocateOutputSetMklShape(context, 0, output_tensor, output_tf_shape,
|
