HomeSort by relevance Sort by last modified time
    Searched refs:permuted_reshaped_padded (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/compiler/tests/
spacetobatch_op_test.py 62 permuted_reshaped_padded = np.transpose(reshaped_padded, (
67 return permuted_reshaped_padded.reshape(output_shape)
  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
spacetobatch_op.cc 108 // `permuted_reshaped_padded` of shape:
124 xla::ComputationDataHandle permuted_reshaped_padded = local
127 // 4. Reshape `permuted_reshaped_padded` to flatten `block_shape` into the
146 b->Reshape(permuted_reshaped_padded, output_shape);
  /external/tensorflow/tensorflow/python/kernel_tests/
spacetobatch_op_test.py 66 permuted_reshaped_padded = np.transpose(reshaped_padded, (
71 return permuted_reshaped_padded.reshape(output_shape)

Completed in 423 milliseconds