Searched defs:d_shape (Results 1 – 3 of 3) sorted by relevance
267 ae_int32x2 d_str, d_shape; in xa_nn_elm_add_broadcast_4D_f32xf32_f32() local
215 const TensorShape d_shape(n, m, n_batches, n_gemms); in configure() local
2371 Shape d_shape = ShapeUtil::MakeShape(indices_type, {2}); in OptimizeDotOfGather() local