Searched refs:GetTensorMetaDataIndex (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 681 ctext->input(GetTensorMetaDataIndex(n, ctext->num_inputs())) in GetMklShape()
684 ctext->input(GetTensorMetaDataIndex(n, ctext->num_inputs())) in GetMklShape()
747 GetTensorMetaDataIndex(n, ctext->num_outputs()), in AllocateOutputSetMklShape()
769 GetTensorMetaDataIndex(n, ctext->num_outputs()),
818 int idx_meta_in = GetTensorMetaDataIndex(idx_in, num_inputs); in CopyMklTensorInToOut()
820 int idx_meta_out = GetTensorMetaDataIndex(idx_out, num_outputs); in CopyMklTensorInToOut()
874 int idx_meta_in = GetTensorMetaDataIndex(idx_in, num_inputs); in ForwardMklTensorInToOut()
876 int idx_meta_out = GetTensorMetaDataIndex(idx_out, num_outputs); in ForwardMklTensorInToOut()
933 GetTensorMetaDataIndex(idx_data_in, context->num_inputs()); in ForwardMklMetaDataInToOut()
935 GetTensorMetaDataIndex(idx_data_out, context->num_outputs()); in ForwardMklMetaDataInToOut()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | mkl_graph_util.h | 74 int inline GetTensorMetaDataIndex(int n, int total_tensors) { in GetTensorMetaDataIndex() function
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | mkl_layout_pass.cc | 2201 GetTensorMetaDataIndex(n_output_slot, n->num_outputs()); in GetNodeProducingMklTensor()
3922 GetTensorMetaDataIndex(n_data_op_slot, n_data->num_outputs()); in FixMklMetaDataEdgeIfNeeded()
3983 GetTensorMetaDataIndex(e->dst_input(), n->num_inputs()); in FixMklMetaDataEdges()
|