OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:SetMklTensor
(Results
1 - 18
of
18
) sorted by null
/external/tensorflow/tensorflow/core/kernels/
mkl_softmax_op.cc
109
output_mkl_shape.
SetMklTensor
(true);
116
output_mkl_shape.
SetMklTensor
(false);
mkl_relu_op.cc
81
mkl_context.output_shape.
SetMklTensor
(false);
123
mkl_context.output_shape.
SetMklTensor
(true);
137
mkl_context.output_shape.
SetMklTensor
(false);
287
mkl_context.output_shape.
SetMklTensor
(false);
336
mkl_context.output_shape.
SetMklTensor
(true);
358
mkl_context.output_shape.
SetMklTensor
(false);
426
dnn_shape_dst.
SetMklTensor
(true);
435
dnn_shape_dst.
SetMklTensor
(false);
583
dnn_shape_diff_src.
SetMklTensor
(true);
592
dnn_shape_diff_src.
SetMklTensor
(false)
[
all
...]
mkl_aggregate_ops.cc
78
mkl_context.output_shape.
SetMklTensor
(false);
99
mkl_context.output_shape.
SetMklTensor
(false);
140
mkl_context.output_shape.
SetMklTensor
(true);
161
mkl_context.output_shape.
SetMklTensor
(false);
337
mkl_shape_dst.
SetMklTensor
(false);
351
mkl_shape_dst.
SetMklTensor
(false);
461
output_mkl_shape.
SetMklTensor
(true);
476
output_mkl_shape.
SetMklTensor
(false);
mkl_fused_batch_norm_op.cc
131
mkl_shape_output.
SetMklTensor
(true);
150
mkl_shape_batch_mean.
SetMklTensor
(false);
156
mkl_shape_batch_variance.
SetMklTensor
(false);
177
mkl_shape_saved_mean.
SetMklTensor
(false);
186
mkl_shape_saved_variance.
SetMklTensor
(false);
418
mkl_shape_in_backprop.
SetMklTensor
(true);
[
all
...]
mkl_reshape_op.cc
113
mkl_shape_output.
SetMklTensor
(false);
254
mkl_shape_output.
SetMklTensor
(false);
286
mkl_shape_output.
SetMklTensor
(true);
mkl_conv_ops.cc
192
mkl_output_mkl_shape.
SetMklTensor
(false);
201
mkl_output_mkl_shape.
SetMklTensor
(false);
276
mkl_output_mkl_shape.
SetMklTensor
(true);
293
mkl_filter_output_mkl_shape.
SetMklTensor
(true);
553
output_mkl_shape.
SetMklTensor
(false);
558
filter_mkl_shape.
SetMklTensor
(false);
675
output_mkl_shape.
SetMklTensor
(true);
699
filter_mkl_shape.
SetMklTensor
(true);
[
all
...]
mkl_input_conversion_op.cc
192
mkl_output_mkl_shape.
SetMklTensor
(true);
317
mkl_output_mkl_shape.
SetMklTensor
(true);
425
mkl_output_mkl_shape.
SetMklTensor
(true);
mkl_lrn_op.cc
219
mkl_output_mkl_shape.
SetMklTensor
(true);
236
mkl_workspace_mkl_shape.
SetMklTensor
(false);
301
mkl_output_mkl_shape.
SetMklTensor
(false);
306
mkl_workspace_mkl_shape.
SetMklTensor
(false);
449
mkl_output_mkl_shape.
SetMklTensor
(true);
671
mkl_output_mkl_shape.
SetMklTensor
(false);
[
all
...]
mkl_concat_op.cc
388
mkl_output_mkl_shape.
SetMklTensor
(true);
514
mkl_tensor_mkl_shape.
SetMklTensor
(false);
545
mkl_tensor_mkl_shape.
SetMklTensor
(false);
751
dnn_shape_dst.
SetMklTensor
(true);
759
dnn_shape_dst.
SetMklTensor
(false);
797
dnn_shape_output.
SetMklTensor
(false);
[
all
...]
mkl_conv_grad_filter_ops.cc
228
mkl_context.grad_filter_shape.
SetMklTensor
(false);
580
output_mkl_shape.
SetMklTensor
(false);
598
bias_grad_mkl_shape.
SetMklTensor
(false);
mkl_pooling_ops_common.h
222
output_mkl_shape.
SetMklTensor
(true);
289
output_mkl_shape.
SetMklTensor
(true);
mkl_conv_grad_bias_ops.cc
89
output_mkl_shape.
SetMklTensor
(false);
mkl_maxpooling_op.cc
106
mkl_out_shape.
SetMklTensor
(true);
124
mkl_workspace_shape.
SetMklTensor
(false);
277
mkl_output_shape.
SetMklTensor
(true);
596
workspace_mkl_shape.
SetMklTensor
(false);
[
all
...]
mkl_conv_grad_input_ops.cc
208
mklOutputShape.
SetMklTensor
(true);
465
output_mkl_shape.
SetMklTensor
(true);
mkl_avgpooling_op.cc
126
mkl_out_shape.
SetMklTensor
(true);
316
mkl_out_shape.
SetMklTensor
(true);
471
output_mkl_shape.
SetMklTensor
(false);
mkl_conv_ops.h
390
output_mkl_shape.
SetMklTensor
(false);
/external/tensorflow/tensorflow/core/util/
mkl_util_test.cc
34
a_mkldnn_shape.
SetMklTensor
(true);
46
b_mkldnn_shape.
SetMklTensor
(true);
mkl_util.h
83
void
SetMklTensor
(const bool isMklTensor) { isMklTensor_ = isMklTensor; }
419
inline void
SetMklTensor
(bool is_mkl_tensor) {
[
all
...]
Completed in 675 milliseconds