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

  /external/tensorflow/tensorflow/core/kernels/
spectrogram_op.cc 63 const int64 output_slices = channel_count; variable
69 0, TensorShape({output_slices, output_height, output_width}),
  /external/tensorflow/tensorflow/compiler/xla/service/
dot_decomposer.cc 101 std::vector<HloInstruction*> output_slices(batch_size);
144 output_slices[i] = dot_r3;
147 // Concatenate slices from 'output_slices' along batch dimension.
149 HloInstruction::CreateConcatenate(concat_shape_r3, output_slices, 0));

Completed in 75 milliseconds