Searched refs:input_in_mkl_format (Results 1 - 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dmkl_identity_op.cc48 bool input_in_mkl_format = mkl_shape_input.IsMklTensor(); variable
50 if (input_in_mkl_format) {
H A Dmkl_reshape_op.cc96 bool input_in_mkl_format = mkl_shape_input.IsMklTensor(); variable
97 if (input_in_mkl_format) {
168 bool input_in_mkl_format = mkl_shape_input.IsMklTensor(); variable
169 const int64 nelems = input_in_mkl_format
219 if (input_in_mkl_format) {
H A Dmkl_maxpooling_op.cc80 bool input_in_mkl_format = mkl_context.input_shape.IsMklTensor(); variable
84 if (input_in_mkl_format == false) {
154 bool input_in_mkl_format = input_shape.IsMklTensor(); local
156 if (input_in_mkl_format == false) {
181 bool input_in_mkl_format = input_shape.IsMklTensor(); local
183 if (!input_in_mkl_format) {
242 bool input_in_mkl_format = mkl_context.input_shape.IsMklTensor(); variable
244 if (input_in_mkl_format == false)
250 if (input_in_mkl_format == false) {
318 bool input_in_mkl_format local
391 bool input_in_mkl_format = input_shape.IsMklTensor(); local
470 bool input_in_mkl_format = input_shape.IsMklTensor(); local
[all...]
H A Dmkl_fused_batch_norm_op.cc70 bool input_in_mkl_format = mkl_context.mkl_shape_input_shape.IsMklTensor(); variable
72 if (!input_in_mkl_format) {
235 bool input_in_mkl_format = mkl_shape_input_shape.IsMklTensor(); local
236 if (!input_in_mkl_format) dnnLayoutDelete_F32(mkl_lt_input);
243 bool input_in_mkl_format = mkl_shape_input_shape.IsMklTensor(); local
244 mkl_params.in_dim = input_in_mkl_format
248 input_in_mkl_format ? mkl_shape_input_shape.GetSizes()[0]
251 input_in_mkl_format ? mkl_shape_input_shape.GetSizes()[1]
254 input_in_mkl_format ? mkl_shape_input_shape.GetSizes()[2]
257 input_in_mkl_format
266 bool input_in_mkl_format = mkl_shape_input_shape.IsMklTensor(); local
371 bool input_in_mkl_format = mkl_context.mkl_shape_input_shape.IsMklTensor(); variable
515 bool input_in_mkl_format = mkl_shape_input_shape.IsMklTensor(); local
526 bool input_in_mkl_format = mkl_shape_input_shape.IsMklTensor(); local
548 bool input_in_mkl_format = mkl_shape_input_shape.IsMklTensor(); local
[all...]
H A Dmkl_conv_ops.cc93 bool input_in_mkl_format = mkl_context.input_shape.IsMklTensor(); variable
108 if (!input_in_mkl_format) {
126 input_in_mkl_format ? GetMklTensorDim(mkl_context.input_shape, 'C')
138 input_in_mkl_format ? GetMklTensorDim(mkl_context.input_shape, 'H')
150 input_in_mkl_format ? GetMklTensorDim(mkl_context.input_shape, 'W')
161 input_in_mkl_format ? GetMklTensorDim(mkl_context.input_shape, 'N')
208 mkl_context.in_dims = input_in_mkl_format
353 bool input_in_mkl_format = input_shape.IsMklTensor(); local
354 if (input_in_mkl_format) {
473 bool input_in_mkl_format local
[all...]
H A Dmkl_conv_grad_filter_ops.cc84 bool input_in_mkl_format = mkl_context.input_shape.IsMklTensor(); variable
107 if (input_in_mkl_format) {
153 mkl_context.in_dims = input_in_mkl_format
291 bool input_in_mkl_format = input_shape.IsMklTensor(); local
293 if (!input_in_mkl_format) dnnLayoutDelete_F32(lt_input);
301 bool input_in_mkl_format = input_shape.IsMklTensor(); local
302 if (input_in_mkl_format) {
H A Dmkl_avgpooling_op.cc72 bool input_in_mkl_format = mkl_context.input_shape.IsMklTensor(); variable
74 if (!input_in_mkl_format)
80 if (!input_in_mkl_format) {
102 if (input_in_mkl_format == false) {
163 bool input_in_mkl_format = input_shape.IsMklTensor(); local
165 if (!input_in_mkl_format) {
201 bool input_in_mkl_format = input_shape.IsMklTensor(); local
202 if (!input_in_mkl_format) {
H A Dmkl_relu_op.cc76 bool input_in_mkl_format = mkl_context.input_shape.IsMklTensor(); variable
78 if (!input_in_mkl_format && !input.dims()) { // handle the case of a scalar
91 if (input_in_mkl_format) {
121 if (input_in_mkl_format) {
162 bool input_in_mkl_format = input_shape.IsMklTensor(); local
163 if (!input_in_mkl_format) {
172 bool input_in_mkl_format = input_shape.IsMklTensor(); local
173 if (!input_in_mkl_format) {
H A Dmkl_lrn_op.cc99 bool input_in_mkl_format = mkl_context.input_shape.IsMklTensor(); variable
102 mkl_context.in_dims = input_in_mkl_format
112 if (!input_in_mkl_format) {
118 if (input_in_mkl_format) {

Completed in 329 milliseconds