Searched refs:GetTensorDataIndex (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/graph/ |
| D | mkl_graph_util.h | 65 int inline GetTensorDataIndex(int n, int total_tensors) { in GetTensorDataIndex() function
77 int tidx = GetTensorDataIndex(n, total_tensors); in GetTensorMetaDataIndex()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 699 return ctext->input(GetTensorDataIndex(n, ctext->num_inputs())); in MklGetInput()
762 ctext, ctext->allocate_output(GetTensorDataIndex(n, ctext->num_outputs()),
817 int idx_data_in = GetTensorDataIndex(idx_in, num_inputs); in CopyMklTensorInToOut()
819 int idx_data_out = GetTensorDataIndex(idx_out, num_outputs); in CopyMklTensorInToOut()
839 int idx_data_in = GetTensorDataIndex(idx_in, num_inputs); in CopyTfTensorInToOutWithShape()
840 int idx_data_out = GetTensorDataIndex(idx_out, num_outputs); in CopyTfTensorInToOutWithShape()
856 int idx_data_in = GetTensorDataIndex(idx_in, num_inputs); in ForwardTfTensorInToOut()
857 int idx_data_out = GetTensorDataIndex(idx_out, num_outputs); in ForwardTfTensorInToOut()
873 int idx_data_in = GetTensorDataIndex(idx_in, num_inputs); in ForwardMklTensorInToOut()
875 int idx_data_out = GetTensorDataIndex(idx_out, num_outputs); in ForwardMklTensorInToOut()
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | mkl_layout_pass.cc | 3554 GetTensorDataIndex(e->src_output(), e->src()->num_outputs()), in RewriteNodeForLayoutPropagation()
|