| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | topk_rewriter.cc | 29 auto match_bitcast_f32 = [](int64_t parameter_number) { in IsNanSafeGt()
45 auto match_bitcast_f32_with_convert = [](int64_t parameter_number) { in IsNanSafeGt()
61 auto match_bitcast_bf16 = [](int64_t parameter_number) { in IsNanSafeGt()
78 auto match_bitcast_bf16_with_convert = [](int64_t parameter_number) { in IsNanSafeGt()
|
| D | hlo_input_output_alias_config.h | 47 : parameter_number(parameter_number), in parameter_number() function
51 int64 parameter_number; member
|
| D | source_map_util.h | 48 Status InvalidParameterArgument(Executable* executable, int parameter_number, in InvalidParameterArgument()
|
| D | hlo_matchers.h | 49 explicit HloParameterMatcher(int64_t parameter_number) in HloParameterMatcher()
325 int64_t parameter_number) { in Parameter()
|
| D | sort_simplifier.cc | 85 int64_t parameter_number = 0; in RemoveUnusedOperandFromSort() local
|
| D | local_service.cc | 81 const XlaComputation& computation, int parameter_number) { in ParameterMetadata()
|
| D | hlo.proto | 104 int64 parameter_number = 9; field
373 int64 parameter_number = 2; field
490 int64 parameter_number = 6; field
|
| D | buffer_assignment.h | 127 int64 parameter_number() const { in parameter_number() function
278 void set_entry_computation_parameter(int64_t parameter_number, in set_entry_computation_parameter()
|
| D | while_loop_all_reduce_code_motion.cc | 126 int parameter_number = in IsAllReduceMovable() local
|
| D | hlo_instructions.h | 1114 int64 parameter_number() const { return parameter_number_; } in parameter_number() function
|
| D | hlo_verifier.cc | 743 const HloComputation* computation, int64_t parameter_number) { in CheckOperandAndParameter()
|
| D | hlo_instruction.cc | 939 int64_t parameter_number, const Shape& shape, const string& name) { in CreateParameter()
4360 int64 HloInstruction::parameter_number() const { in parameter_number() function in xla::HloInstruction
|
| D | hlo_parser.cc | 1096 int64_t parameter_number; in ParseInstructionRhs() local
|
| D | hlo_instructions.cc | 2043 HloParameterInstruction::HloParameterInstruction(int64_t parameter_number, in HloParameterInstruction()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | client_library_test_base.h | 606 NativeT value, int64_t parameter_number, const string& name, in CreateR0Parameter()
620 absl::Span<const NativeT> values, int64_t parameter_number, in CreateR1Parameter()
634 const Array2D<NativeT>& array_2d, int64_t parameter_number, in CreateR2Parameter()
648 const Array3D<NativeT>& array_3d, int64_t parameter_number, in CreateR3Parameter()
662 const Array4D<NativeT>& array_4d, int64_t parameter_number, in CreateR4Parameter()
676 const Array<NativeT>& array, int64_t parameter_number, const string& name, in CreateParameter()
|
| D | client_library_test_base.cc | 610 int64_t parameter_number, const Literal& literal, const string& name, in CreateParameterAndTransferLiteral()
640 int64_t parameter_number, const Literal& literal, const string& name, in CreateParameterAndTransferLiteral()
|
| /external/clang/test/SemaCXX/ |
| D | addr-of-overloaded-function.cpp | 182 void parameter_number() { in parameter_number() function
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 428 XlaOp Parameter(int64_t parameter_number, const Shape& shape, in Parameter()
|
| D | xla_builder.cc | 855 int64_t parameter_number, const Shape& shape, const string& name, in Parameter()
3891 XlaOp Parameter(XlaBuilder* builder, int64_t parameter_number, in Parameter()
3897 XlaOp Parameter(XlaBuilder* builder, int64_t parameter_number, in Parameter()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_util.cc | 1157 auto match_bitcast_f32 = [](int64_t parameter_number) { in IsNanSafeGt()
1172 auto match_bitcast_bf16 = [](int64_t parameter_number) { in IsNanSafeGt()
|