Searched defs:output_shape_vec (Results 1 – 8 of 8) sorted by relevance
114 auto output_shape_vec = output_shape.flat<Index>(); in Compute() local220 auto output_shape_vec = output_shape.flat<Index>(); in ComputeAsync() local
170 auto output_shape_vec = result_shape->vec<int64_t>(); in ReshapeSparseTensor() local
111 auto output_shape_vec = output_shape->vec<int64_t>(); in Compute() local
181 at::SymDimVector output_shape_vec(output_shape); in qembeddingbag_byte_unpack_meta() local
343 at::SymDimVector output_shape_vec(output_shape); in qembeddingbag_byte_prepack_meta() local
155 auto output_shape_vec = output_shape.vec<int64_t>(); in Compute() local
294 auto output_shape_vec = output_shape_t.vec<int32>(); in Compute() local
2326 std::vector<Shape> output_shape_vec; in InferReduceWindowShape() local