OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:GetTfShape
(Results
1 - 12
of
12
) sorted by null
/external/tensorflow/tensorflow/core/util/
mkl_util_test.cc
39
TensorShape a_mkldnn_tf_shape = a_mkldnn_shape.
GetTfShape
();
51
TensorShape b_mkldnn_tf_shape = b_mkldnn_shape.
GetTfShape
();
mkl_util.h
400
return this->
GetTfShape
() == input_shape.
GetTfShape
() &&
415
return this->
GetTfShape
() == input_shape;
477
inline TensorShape
GetTfShape
() const {
786
inline TensorShape
GetTfShape
(OpKernelContext* context, size_t input_idx) {
794
return input_mkl_shape.
GetTfShape
();
[
all
...]
/external/tensorflow/tensorflow/core/kernels/
mkl_softmax_op.cc
64
? src_mkl_shape.
GetTfShape
()
mkl_conv_ops.h
189
GetFilterSizeInMklOrder(
GetTfShape
(context_, src_index),
190
GetTfShape
(context_, filter_index), filter_dims);
282
auto input_tf_shape =
GetTfShape
(context_, src_index);
283
auto filter_tf_shape =
GetTfShape
(context_, filter_index);
382
TensorShape outbprop_tf_shape =
GetTfShape
(context, kOutbpropIdx);
mkl_tfconv_op.h
102
TensorShape output_shape = input_shape.
GetTfShape
();
mkl_reshape_op.cc
170
? mkl_shape_input.
GetTfShape
().num_elements()
221
TensorShape shape_from = mkl_shape_input.
GetTfShape
();
mkl_concat_op.cc
609
? input_shapes[0].
GetTfShape
()
622
s.IsMklTensor() ? s.
GetTfShape
() : input_tensors[i].shape();
667
s.IsMklTensor() ? s.
GetTfShape
() : input_tensors[i].shape();
677
dst_dims = TFShapeToMklDnnDims(input_shapes[0].
GetTfShape
());
693
? TFShapeToMklDnnDims(input_shapes[k].
GetTfShape
())
[
all
...]
mkl_fused_batch_norm_op.cc
[
all
...]
mkl_aggregate_ops.cc
320
src1_shape = input1_in_mkl_format ? src1_mkl_shape.
GetTfShape
()
322
src2_shape = input2_in_mkl_format ? src2_mkl_shape.
GetTfShape
()
mkl_conv_grad_filter_ops.cc
459
return
GetTfShape
(context, input_idx);
512
TensorShape obp_tf_shape =
GetTfShape
(context, 2);
mkl_conv_ops.cc
536
auto src_tf_shape =
GetTfShape
(context, kInputIndex_Src);
537
auto filter_tf_shape =
GetTfShape
(context, kInputIndex_Filter);
[
all
...]
mkl_conv_grad_input_ops.cc
397
return
GetTfShape
(context, kInputIndex_Filter);
Completed in 966 milliseconds