Home
last modified time | relevance | path

Searched defs:parameter_number (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_input_output_alias_config.h47 : parameter_number(parameter_number), in parameter_number() function
51 int64 parameter_number; member
Dtopk_rewriter.cc29 auto match_bitcast_f32 = [](int64 parameter_number) { in IsNanSafeGt()
44 auto match_bitcast_bf16 = [](int64 parameter_number) { in IsNanSafeGt()
Dsource_map_util.h48 Status InvalidParameterArgument(Executable* executable, int parameter_number, in InvalidParameterArgument()
Dhlo_matchers.h46 explicit HloParameterMatcher(int64 parameter_number) in HloParameterMatcher()
291 int64 parameter_number) { in Parameter()
Dsort_simplifier.cc85 int64 parameter_number = 0; in RemoveUnusedOperandFromSort() local
Dlocal_service.cc81 const XlaComputation& computation, int parameter_number) { in ParameterMetadata()
Dhlo.proto69 int64 parameter_number = 9; field
328 int64 parameter_number = 2; field
442 int64 parameter_number = 6; field
Dbuffer_assignment.h127 int64 parameter_number() const { in parameter_number() function
277 void set_entry_computation_parameter(int64 parameter_number, in set_entry_computation_parameter()
Dwhile_loop_all_reduce_code_motion.cc126 int parameter_number = in IsAllReduceMovable() local
Dhlo_instructions.h1047 int64 parameter_number() const { return parameter_number_; } in parameter_number() function
Dhlo_verifier.cc409 const HloComputation* computation, int64 parameter_number) { in CheckOperandAndParameter()
Dhlo_instruction.cc824 int64 parameter_number, const Shape& shape, const string& name) { in CreateParameter()
4062 int64 HloInstruction::parameter_number() const { in parameter_number() function in xla::HloInstruction
Dhlo_parser.cc1038 int64 parameter_number; in ParseInstructionRhs() local
Dhlo_instructions.cc1905 HloParameterInstruction::HloParameterInstruction(int64 parameter_number, in HloParameterInstruction()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dclient_library_test_base.h606 NativeT value, int64 parameter_number, const string& name, in CreateR0Parameter()
620 absl::Span<const NativeT> values, int64 parameter_number, in CreateR1Parameter()
634 const Array2D<NativeT>& array_2d, int64 parameter_number, in CreateR2Parameter()
648 const Array3D<NativeT>& array_3d, int64 parameter_number, in CreateR3Parameter()
662 const Array4D<NativeT>& array_4d, int64 parameter_number, in CreateR4Parameter()
676 const Array<NativeT>& array, int64 parameter_number, const string& name, in CreateParameter()
Dclient_library_test_base.cc609 ClientLibraryTestBase::CreateParameterAndTransferLiteral(int64 parameter_number, in CreateParameterAndTransferLiteral()
642 int64 parameter_number, const Literal& literal, const string& name, in CreateParameterAndTransferLiteral()
/external/llvm-project/clang/test/SemaCXX/
Daddr-of-overloaded-function.cpp182 void parameter_number() { in parameter_number() function
/external/clang/test/SemaCXX/
Daddr-of-overloaded-function.cpp182 void parameter_number() { in parameter_number() function
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h445 XlaOp Parameter(int64 parameter_number, const Shape& shape, in Parameter()
Dxla_builder.cc845 int64 parameter_number, const Shape& shape, const string& name, in Parameter()
4079 XlaOp Parameter(XlaBuilder* builder, int64 parameter_number, const Shape& shape, in Parameter()
4085 XlaOp Parameter(XlaBuilder* builder, int64 parameter_number, const Shape& shape, in Parameter()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.cc1111 auto match_bitcast_f32 = [](int64 parameter_number) { in IsNanSafeGt()
1126 auto match_bitcast_bf16 = [](int64 parameter_number) { in IsNanSafeGt()