| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | propagate_fake_quant_num_bits.cc | 106 case OperatorType::kSelect: in DoesOpBlockBackwardPropagation()
126 case OperatorType::kSelect: in DoesOpInputBlockBackwardPropagation()
|
| D | resolve_constant_select.cc | 36 if (base_op->type != OperatorType::kSelect) { in Run()
|
| D | propagate_array_data_types.cc | 184 case OperatorType::kSelect: { in Run()
|
| D | hardcode_min_max.cc | 494 case OperatorType::kSelect: in Run()
|
| D | quantize.cc | 92 OperatorType::kSelect, in SupportsQuantization()
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | op_version.cc | 225 {{OperatorType::kSelect, 1}, "1.14.0"}, in GetMinimumRuntimeVersionForModel()
226 {{OperatorType::kSelect, 2}, "1.14.0"}, in GetMinimumRuntimeVersionForModel()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | reshape_mover_test.cc | 202 root_shape, HloOpcode::kSelect, const0, reshape1, reshape2));
283 root_shape, HloOpcode::kSelect, pred, reshape0, reshape1));
435 root_shape, HloOpcode::kSelect, reshape_pred, reshape0, reshape1)); in TEST_F()
465 root_shape, HloOpcode::kSelect, reshape_pred, param0, param1)); in TEST_F()
|
| D | bfloat16_support.cc | 111 case HloOpcode::kSelect: in EffectiveOperandPrecisionIsOutputPrecision()
|
| D | comparison_expander.cc | 59 HloInstruction::CreateTernary(signed_shape, HloOpcode::kSelect, in BitcastConvertFloatingPointToIntegral()
|
| D | hlo_opcode.h | 134 V(kSelect, "select", 3) \
|
| D | stable_sort_expander.cc | 192 ShapeUtil::MakeShape(PRED, {}), HloOpcode::kSelect, same, tie_breaker, in ExpandInstruction()
|
| D | stable_sort_expander_test.cc | 67 ASSERT_EQ(root->opcode(), HloOpcode::kSelect); in CheckComputationHasTieBreaker()
|
| D | convolution_group_converter.cc | 348 expanded_filter_shape, HloOpcode::kSelect, filter_mask, new_convolution, in HandleBatchGroupCount()
477 expanded_filter_shape, HloOpcode::kSelect, filter_mask, in HandleConvolution()
|
| D | shape_inference_test.cc | 116 HloOpcode::kSelect, pred_, tuple, tuple); in TEST_F()
124 HloOpcode::kSelect, pred_, matrix_64_48_, matrix_64_48_); in TEST_F()
134 HloOpcode::kSelect, predarray, matrix_64_48_, matrix_64_48_); in TEST_F()
141 HloOpcode::kSelect, pred_, matrix_64_48_, matrix_32_64_); in TEST_F()
147 HloOpcode::kSelect, s32_, matrix_64_48_, matrix_64_48_); in TEST_F()
153 HloOpcode::kSelect, ShapeUtil::MakeShape(PRED, {64}), matrix_64_48_, in TEST_F()
162 HloOpcode::kSelect, ShapeUtil::MakeTupleShape({pred_, pred_}), in TEST_F()
|
| D | conditional_simplifier.cc | 180 f->shape(), HloOpcode::kSelect, condition_broadcast(f->shape()), in TryRemoveConditional()
|
| D | instruction_fusion.cc | 109 case HloOpcode::kSelect: in IsExpensive()
|
| D | algebraic_simplifier.cc | 589 case HloOpcode::kSelect: in IsOpCodeMultiplyCommutative()
1490 a->shape(), HloOpcode::kSelect, dividend_is_negative, in TryDivideToShift()
1501 return HloInstruction::CreateTernary(divide->shape(), HloOpcode::kSelect, in TryDivideToShift()
2557 HloInstruction::CreateTernary(gather->shape(), HloOpcode::kSelect, in HandleGather()
2741 multiply->shape(), HloOpcode::kSelect, convert_operand, in HandleMultiply()
3714 a->shape(), HloOpcode::kSelect, dividend_is_negative, in TryRemainderToAnd()
3726 remainder->shape(), HloOpcode::kSelect, dividend_is_negative, in TryRemainderToAnd()
|
| D | hlo_instruction.cc | 960 case HloOpcode::kSelect: in CreateTernary()
1766 case HloOpcode::kSelect: in CloneWithNewOperands()
2162 case HloOpcode::kSelect: in IdenticalSlowPath()
2663 case HloOpcode::kSelect: in IsOpElementwise()
3122 case HloOpcode::kSelect: in Visit()
|
| D | hlo_creation_utils.cc | 403 op_shape.IsTuple() ? HloOpcode::kTupleSelect : HloOpcode::kSelect; in MakeSelectHlo()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cusolver_rewriter.cc | 125 a_shape, HloOpcode::kSelect, ok, out, nans)); in CreateCholesky()
|
| D | gpu_fusible_test.cc | 89 ASSERT_EQ(loop_fusion->fused_expression_root()->opcode(), HloOpcode::kSelect); in TEST_F()
127 ASSERT_EQ(loop_fusion->fused_expression_root()->opcode(), HloOpcode::kSelect); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | gather_scatter_handler.cc | 283 pgather->shape(), HloOpcode::kSelect, broadcast_filter, in ParititonTrivialIndexedOperandDimension()
597 identity->shape(), HloOpcode::kSelect, not_partition_zero, in HandleScatter()
|
| D | spmd_partitioner.cc | 540 shape, HloOpcode::kSelect, mask, hlo_, broadcast_pad_value)); in PadWithValue()
1076 shape, HloOpcode::kSelect, is_src_core, hlo(), zero_bcast)); in Broadcast()
2430 new_indices[dim]->shape(), HloOpcode::kSelect, in HandleDynamicUpdateSlice()
2445 dus->shape(), HloOpcode::kSelect, in HandleDynamicUpdateSlice()
3356 zero->shape(), HloOpcode::kSelect, is_shard0, pad_low_hlo, in HandleSelectAndScatter()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | model.h | 150 kSelect, enumerator
1357 SelectOperator() : Operator(OperatorType::kSelect) {}
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_utils.cc | 461 return xla::HloOpcode::kSelect; in MhloToHloOpcode()
|