Home	Sort by relevance Sort by last modified time

Full Search Definition Symbol File Path History   | | Help

    Searched full:input_in_mkl_format (Results 1 - 9 of 9) sorted by null

/external/tensorflow/tensorflow/core/kernels/
mkl_identity_op.cc	48 bool input_in_mkl_format = mkl_shape_input.IsMklTensor(); variable 50 if (input_in_mkl_format) {
mkl_maxpooling_op.cc	80 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 = input_shape.IsMklTensor() local 391 bool input_in_mkl_format = input_shape.IsMklTensor(); local 470 bool input_in_mkl_format = input_shape.IsMklTensor(); local     [all...]
mkl_fused_batch_norm_op.cc	70 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 ? mkl_shape_input_shape.GetSizes()[3 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...]
mkl_conv_ops.cc	93 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 = input_shape.IsMklTensor() local     [all...]
mkl_avgpooling_op.cc	72 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) {
mkl_reshape_op.cc	96 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) {
mkl_conv_grad_filter_ops.cc	84 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) {
mkl_relu_op.cc	76 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) {     [all...]
mkl_lrn_op.cc	99 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) {     [all...]

Completed in 656 milliseconds