| /external/tensorflow/tensorflow/core/util/ |
| mkl_util.h | 838 const TensorShape& tf_shape,
845 tf_shape, output));
859 const TensorShape& tf_shape,
866 tf_shape, output));
882 TensorShape tf_shape; local
893 TensorShape tf_shape; local
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_relu_op.cc | 122 TensorShape tf_shape; variable
130 tf_shape.AddDim(dnnLayoutGetMemorySize_F32(static_cast<dnnLayout_t>(
133 AllocateOutputSetMklShape(context, 0, &output, tf_shape,
335 TensorShape tf_shape; local
351 tf_shape.AddDim(dnnLayoutGetMemorySize_F32(static_cast<dnnLayout_t>(
354 AllocateOutputSetMklShape(context, 0, &output, tf_shape,
[all...] |
| mkl_aggregate_ops.cc | 139 TensorShape tf_shape; variable
153 tf_shape.AddDim(dnnLayoutGetMemorySize_F32(static_cast<dnnLayout_t>(
157 AllocateOutputSetMklShape(ctx, src1_idx, &output, tf_shape,
|
| /external/tensorflow/tensorflow/python/layers/ |
| normalization_test.py | [all...] |
