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

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
depthtospace_op.cc 136 // `permuted_reshaped` of shape:
144 xla::ComputationDataHandle permuted_reshaped = variable
147 // 3. Reshape `permuted_reshaped` to flatten `block_shape` into the
156 b->Reshape(permuted_reshaped, output_shape);
spacetodepth_op.cc 141 // `permuted_reshaped` of shape:
148 xla::ComputationDataHandle permuted_reshaped = variable
151 // 3. Reshape `permuted_reshaped` to flatten `block_shape` into the
160 b->Reshape(permuted_reshaped, output_shape);

Completed in 357 milliseconds