Searched refs:kForward (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_cluster_util.cc | 418 enum class Direction { kForward, kBackward }; enumerator
484 if (direction == Direction::kForward) { in GetNodesRelatedToRefVariablesInDirection()
524 direction == Direction::kForward ? n->in_edges() : n->out_edges(); in GetNodesRelatedToRefVariablesInDirection()
526 if (result->contains(direction == Direction::kForward ? e->src() in GetNodesRelatedToRefVariablesInDirection()
538 if (direction == Direction::kForward && in GetNodesRelatedToRefVariablesInDirection()
565 graph, lib_runtime, Direction::kForward, 0, &result)); in GetNodesRelatedToRefVariables()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_pad_for_convolutions.cc | 210 case CudnnConvKind::kForward: in TryResolvePaddedShapesForTensorCore()
294 if (kind != CudnnConvKind::kForward && in TryResolvePaddedShapesForIntegerConvolution()
315 case CudnnConvKind::kForward: in TryResolvePaddedShapesForIntegerConvolution()
|
| D | gpu_conv_runner.cc | 157 case CudnnConvKind::kForward: in RunGpuConvInternalImpl()
205 case CudnnConvKind::kForward: in RunGpuConvInternalImpl()
281 case CudnnConvKind::kForward: in GetGpuConvConfig()
465 case CudnnConvKind::kForward: in GetGpuConvParams()
|
| D | ir_emission_utils.h | 54 kForward, // input + filter => output enumerator
|
| D | gpu_layout_assignment.cc | 133 case CudnnConvKind::kForward: in AddBackendConstraintsToDnnConvCustomCall()
|
| D | ir_emission_utils.cc | 556 return CudnnConvKind::kForward; in GetCudnnConvKind()
572 case CudnnConvKind::kForward: in CudnnConvKindToString()
|
| D | stream_executor_util.cc | 329 case CudnnConvKind::kForward: in GetDNNConvKindFromCudnnConvKind()
|
| D | gpu_conv_padding_legalization.cc | 413 case CudnnConvKind::kForward: in RunOnComputation()
|
| D | gpu_conv_algorithm_picker.cc | 114 case CudnnConvKind::kForward: in GetAlgorithms()
|
| D | ir_emitter_unnested.cc | 1195 descriptor.kind = CudnnConvKind::kForward; in EmitConvolutionThunkFromMlir()
|
| /external/webrtc/modules/audio_processing/ |
| D | audio_processing_unittest.cc | 81 enum StreamDirection { kForward = 0, kReverse }; enumerator
505 num_reverse_channels, num_reverse_channels, kForward); in Init()
1961 num_reverse_input_channels, num_reverse_output_channels, kForward) in ProcessFormat()
2059 stream_directions.push_back(kForward); in TEST_P()
|
| /external/angle/third_party/abseil-cpp/absl/strings/internal/ |
| D | cord_rep_ring.cc | 40 enum class Direction { kForward, kReversed }; enumerator
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 804 case xla::gpu::CudnnConvKind::kForward: { in EmitDnnConvolution()
|