Searched defs:dest_shape (Results 1 – 5 of 5) sorted by relevance
1295 TensorShape dest_shape; in ComputeWithReservedSpace() local1540 TensorShape dest_shape; in ComputeWithReservedSpace() local
202 const Shape& source_shape, const Shape& dest_shape, in StrideConfig()556 absl::Span<const NativeT> src, const Shape& dest_shape, in CopyElementsBetween()1767 auto dest_shape = subshape(); in CopyElementsWithDynamicBound() local
990 Shape dest_shape = ToShape(*dest); in operator ()() local1060 const Shape& dest_shape = in operator ()() local
3195 const Shape dest_shape = GetShape(dest); in TryBuildConstantInitializerThunk() local3219 const Shape dest_shape = GetShape(dest); in BuildInitializerThunk() local3261 const Shape dest_shape = GetShape(dest); in BuildFusedInitializerThunk() local
1511 const Shape& dest_shape = ShapeUtil::GetTupleElementShape(operand_shape, 0); in HandleCopyDone() local