OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:TfDimIdx
(Results
1 - 2
of
2
) sorted by null
/external/tensorflow/tensorflow/core/util/
mkl_util.h
483
shape[idx] = data_.sizes_[
TfDimIdx
(idx)];
582
inline size_t
TfDimIdx
(int index) const { return data_.map_[index]; }
584
return data_.sizes_[
TfDimIdx
(index)];
590
return
TfDimIdx
(d) == MklDnnDims::Dim_C;
595
return
TfDimIdx
(d) == MklDnnDims::Dim_N;
600
return
TfDimIdx
(d) == MklDnnDims::Dim_W;
605
return
TfDimIdx
(d) == MklDnnDims::Dim_H;
[
all
...]
/external/tensorflow/tensorflow/core/kernels/
mkl_concat_op.cc
743
if (are_all_mkl_inputs) concat_dim = input_shapes[0].
TfDimIdx
(concat_dim);
[
all
...]
Completed in 359 milliseconds