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