Searched defs:diag_index_tensor (Results 1 – 1 of 1) sorted by relevance
1336 Status ReadDiagIndex(InferenceContext* c, const Tensor* diag_index_tensor, in ReadDiagIndex()1365 const Tensor* diag_index_tensor = c->input_tensor(1); in MatrixDiagPartV2Shape() local1422 const Tensor* diag_index_tensor = c->input_tensor(1); in MatrixDiagV2Shape() local1522 const Tensor* diag_index_tensor = c->input_tensor(2); in MatrixSetDiagV2Shape() local