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

  /external/tensorflow/tensorflow/compiler/xla/client/
padding.cc 122 int64 output_dimension = local
125 std::max<int64>((output_dimension - 1) * window_stride +
  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_verifier.cc 484 int64 output_dimension = broadcast->dimensions()[operand_dimension];
485 TF_RET_CHECK((output_dimension < broadcast->shape().rank()) &&
486 output_dimension >= 0 &&
487 (broadcast->shape().dimensions(output_dimension) ==
    [all...]

Completed in 558 milliseconds