Searched refs:GetTensorMetaDataIndex (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | mkl_util.h | 684 ctext->input(GetTensorMetaDataIndex(n, ctext->num_inputs())) in GetMklShape()
687 ctext->input(GetTensorMetaDataIndex(n, ctext->num_inputs())) in GetMklShape()
757 GetTensorMetaDataIndex(n, ctext->num_outputs()), in AllocateOutputSetMklShape()
779 GetTensorMetaDataIndex(n, ctext->num_outputs()),
859 int idx_meta_in = GetTensorMetaDataIndex(idx_in, num_inputs); in CopyMklTensorInToOut()
861 int idx_meta_out = GetTensorMetaDataIndex(idx_out, num_outputs); in CopyMklTensorInToOut()
915 int idx_meta_in = GetTensorMetaDataIndex(idx_in, num_inputs); in ForwardMklTensorInToOut()
917 int idx_meta_out = GetTensorMetaDataIndex(idx_out, num_outputs); in ForwardMklTensorInToOut()
974 GetTensorMetaDataIndex(idx_data_in, context->num_inputs()); in ForwardMklMetaDataInToOut()
976 GetTensorMetaDataIndex(idx_data_out, context->num_outputs()); in ForwardMklMetaDataInToOut()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/graph/ |
| H A D | mkl_graph_util.h | 74 int inline GetTensorMetaDataIndex(int n, int total_tensors) { in GetTensorMetaDataIndex() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | mkl_layout_pass.cc | 2161 GetTensorMetaDataIndex(n_output_slot, n->num_outputs()); in GetNodeProducingMklTensor()
3902 GetTensorMetaDataIndex(n_data_op_slot, n_data->num_outputs()); in FixMklMetaDataEdgeIfNeeded()
3963 GetTensorMetaDataIndex(e->dst_input(), n->num_inputs()); in FixMklMetaDataEdges()
|