OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:GetMklLayout
(Results
1 - 16
of
16
) sorted by null
/external/tensorflow/tensorflow/core/kernels/
mkl_pooling_ops_common.h
204
? input_mkl_shape.
GetMklLayout
()
346
? original_input_mkl_shape.
GetMklLayout
()
358
? original_output_mkl_shape.
GetMklLayout
()
371
? input_gradient_mkl_shape.
GetMklLayout
()
mkl_softmax_op.cc
80
? src_mkl_shape.
GetMklLayout
()
mkl_aggregate_ops.cc
154
mkl_context.output_shape.
GetMklLayout
())) /
390
md1 = src1_mkl_shape.
GetMklLayout
();
407
md2 = src2_mkl_shape.
GetMklLayout
();
411
md1 = src1_mkl_shape.
GetMklLayout
();
412
md2 = src2_mkl_shape.
GetMklLayout
();
mkl_relu_op.cc
131
mkl_context.output_shape.
GetMklLayout
())) /
352
mkl_context.output_shape.
GetMklLayout
())) /
404
src_md = dnn_shape_src.
GetMklLayout
();
523
src_md = dnn_shape_src.
GetMklLayout
();
535
diff_dst_md = dnn_shape_diff_dst.
GetMklLayout
();
548
src_md = dnn_shape_src.
GetMklLayout
();
549
diff_dst_md = dnn_shape_diff_dst.
GetMklLayout
();
[
all
...]
mkl_input_conversion_op.cc
297
auto input0_md = input_shape_0.
GetMklLayout
();
298
auto input1_md = input_shape_1.
GetMklLayout
();
431
auto output_mkl_md = mkl_shape->
GetMklLayout
();
mkl_tfconv_op.h
93
auto input_mkl_md = input_shape.
GetMklLayout
();
mkl_reshape_op.cc
153
auto input_mkl_md = mkl_shape_input.
GetMklLayout
();
244
auto input_mkl_md = mkl_shape_input.
GetMklLayout
();
mkl_conv_ops.h
430
? input_mkl_shape.
GetMklLayout
()
435
? filter_mkl_shape.
GetMklLayout
()
467
auto outbprop_md = outbprop_mkl_shape.
GetMklLayout
();
mkl_concat_op.cc
398
static_cast<dnnLayout_t>(mkl_output_mkl_shape.
GetMklLayout
())) /
698
is_mkl_tensor ? input_shapes[k].
GetMklLayout
() :
725
(memory::format)input_shapes[0].
GetMklLayout
().data.format);
767
dnn_shape_dst.IsMklTensor() ? dnn_shape_dst.
GetMklLayout
() : dst_md;
[
all
...]
mkl_conv_ops.cc
286
static_cast<dnnLayout_t>(mkl_output_mkl_shape.
GetMklLayout
())) /
306
mkl_filter_output_mkl_shape.
GetMklLayout
())) /
575
? src_mkl_shape.
GetMklLayout
()
581
? filter_mkl_shape.
GetMklLayout
()
[
all
...]
mkl_fused_batch_norm_op.cc
140
mkl_shape_output.
GetMklLayout
())) /
428
static_cast<dnnLayout_t>(mkl_shape_in_backprop.
GetMklLayout
())) /
[
all
...]
mkl_avgpooling_op.cc
134
mkl_out_shape.
GetMklLayout
())) /
325
mkl_out_shape.
GetMklLayout
())) /
mkl_maxpooling_op.cc
115
mkl_out_shape.
GetMklLayout
())) /
287
static_cast<dnnLayout_t>(mkl_output_shape.
GetMklLayout
())) /
[
all
...]
mkl_lrn_op.cc
227
static_cast<dnnLayout_t>(mkl_output_mkl_shape.
GetMklLayout
())) /
245
mkl_workspace_mkl_shape.
GetMklLayout
())) /
463
static_cast<dnnLayout_t>(mkl_output_mkl_shape.
GetMklLayout
())) /
[
all
...]
mkl_conv_grad_input_ops.cc
218
mklOutputShape.
GetMklLayout
())) /
/external/tensorflow/tensorflow/core/util/
mkl_util.h
151
const dnnLayout_t
GetMklLayout
() const { return mklLayout_; }
401
CompareMklDnnLayouts(this->
GetMklLayout
(),
402
input_shape.
GetMklLayout
());
513
inline const memory::desc
GetMklLayout
() const {
552
return IsMklTensor() ?
GetMklLayout
() : GetTfLayout();
[
all
...]
Completed in 244 milliseconds