Searched refs:SubtleMustCopyFlat (Results 1 – 3 of 3) sorted by relevance
70 internal::spacetobatch::SubtleMustCopyFlat(orig_block_shape, &block_shape); in BatchToSpaceOpCompute()71 internal::spacetobatch::SubtleMustCopyFlat(orig_crops, &crops); in BatchToSpaceOpCompute()
73 internal::spacetobatch::SubtleMustCopyFlat(orig_block_shape, &block_shape); in SpaceToBatchOpCompute()74 internal::spacetobatch::SubtleMustCopyFlat(orig_paddings, &paddings); in SpaceToBatchOpCompute()
68 void SubtleMustCopyFlat(const Tensor& t, OutputType* output) { in SubtleMustCopyFlat() function