Home
last modified time | relevance | path

Searched refs:SubtleMustCopyFlat (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dbatchtospace_op.cc70 internal::spacetobatch::SubtleMustCopyFlat(orig_block_shape, &block_shape); in BatchToSpaceOpCompute()
71 internal::spacetobatch::SubtleMustCopyFlat(orig_crops, &crops); in BatchToSpaceOpCompute()
Dspacetobatch_op.cc73 internal::spacetobatch::SubtleMustCopyFlat(orig_block_shape, &block_shape); in SpaceToBatchOpCompute()
74 internal::spacetobatch::SubtleMustCopyFlat(orig_paddings, &paddings); in SpaceToBatchOpCompute()
Dspacetobatch_functor.h68 void SubtleMustCopyFlat(const Tensor& t, OutputType* output) { in SubtleMustCopyFlat() function