OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:GetCurLayout
(Results
1 - 13
of
13
) sorted by null
/external/tensorflow/tensorflow/core/kernels/
mkl_lrn_op.cc
123
static_cast<dnnLayout_t>(mkl_context.input_shape.
GetCurLayout
());
519
lt_input = static_cast<dnnLayout_t>(ingrad_shape.
GetCurLayout
());
527
lt_output = static_cast<dnnLayout_t>(inimage_shape.
GetCurLayout
());
593
static_cast<dnnLayout_t>(inimage_shape.
GetCurLayout
()))) {
604
static_cast<dnnLayout_t>(inimage_shape.
GetCurLayout
()))) {
607
static_cast<dnnLayout_t>(inimage_shape.
GetCurLayout
()),
796
memory::desc src_md = src_dnn_shape.
GetCurLayout
();
[
all
...]
mkl_input_conversion_op.cc
199
mkl_output_mkl_shape.SetMklLayout(mkl_shape->
GetCurLayout
());
210
mkl_shape->
GetCurLayout
(),
mkl_conv_grad_bias_ops.cc
169
lt_input = static_cast<dnnLayout_t>(input_shape.
GetCurLayout
());
mkl_aggregate_ops.cc
193
lt_input1 = static_cast<dnnLayout_t>(input1_shape.
GetCurLayout
());
201
lt_input2 = static_cast<dnnLayout_t>(input2_shape.
GetCurLayout
());
mkl_conv_grad_filter_ops.cc
303
lt_input = static_cast<dnnLayout_t>(input_shape.
GetCurLayout
());
312
static_cast<dnnLayout_t>(out_backprop_shape.
GetCurLayout
());
mkl_conv_grad_input_ops.cc
257
lt_filter = (dnnLayout_t)filter_shape.
GetCurLayout
();
265
lt_outbackprop = (dnnLayout_t)outback_shape.
GetCurLayout
();
mkl_maxpooling_op.cc
161
lt_user_input = (dnnLayout_t)input_shape.
GetCurLayout
();
326
lt_input_user = (dnnLayout_t)input_shape.
GetCurLayout
();
336
lt_outbackprop_user = (dnnLayout_t)output_backprop_shape.
GetCurLayout
();
[
all
...]
mkl_relu_op.cc
177
lt_input = static_cast<dnnLayout_t>(input_shape.
GetCurLayout
());
238
lt_input = static_cast<dnnLayout_t>(input_shape.
GetCurLayout
());
245
lt_grad = static_cast<dnnLayout_t>(grad_shape.
GetCurLayout
());
[
all
...]
mkl_avgpooling_op.cc
170
lt_user_input = (dnnLayout_t)input_shape.
GetCurLayout
();
385
lt_user_outbackprop = (dnnLayout_t)out_backprop_shape.
GetCurLayout
();
mkl_fused_batch_norm_op.cc
269
static_cast<dnnLayout_t>(mkl_shape_input_shape.
GetCurLayout
());
551
static_cast<dnnLayout_t>(mkl_shape_input_shape.
GetCurLayout
());
562
static_cast<dnnLayout_t>(mkl_shape_out_backprop.
GetCurLayout
());
[
all
...]
mkl_concat_op.cc
437
lt_inputs.push_back((dnnLayout_t)is.
GetCurLayout
());
[
all
...]
mkl_conv_ops.cc
355
lt_input = static_cast<dnnLayout_t>(input_shape.
GetCurLayout
());
[
all
...]
/external/tensorflow/tensorflow/core/util/
mkl_util.h
153
const dnnLayout_t
GetCurLayout
() const {
551
inline const memory::desc
GetCurLayout
() const {
[
all
...]
Completed in 189 milliseconds