Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dtopk_rewriter.cc30 auto match_bitcast_f32 = [](int64_t parameter_number) { in IsNanSafeGt()
46 auto match_bitcast_f32_with_convert = [](int64_t parameter_number) { in IsNanSafeGt()
62 auto match_bitcast_bf16 = [](int64_t parameter_number) { in IsNanSafeGt()
79 auto match_bitcast_bf16_with_convert = [](int64_t parameter_number) { in IsNanSafeGt()
96 auto match_s32 = [](int64_t parameter_number) { in IsNanSafeGt()
H A Dhlo_input_output_alias_config.h47 : parameter_number(parameter_number), in parameter_number() function
51 int64_t parameter_number; member
H A Dhlo_matchers.h49 explicit HloParameterMatcher(int64_t parameter_number) in HloParameterMatcher()
342 int64_t parameter_number) { in Parameter()
H A Dhlo.proto124 int64 parameter_number = 9; field
411 int64 parameter_number = 2; field
563 int64 parameter_number = 6; field
H A Dbuffer_assignment.h125 int64_t parameter_number() const { in parameter_number() function
276 void set_entry_computation_parameter(int64_t parameter_number, in set_entry_computation_parameter()
H A Dsource_map_util.h47 Status InvalidParameterArgument(Executable* executable, int parameter_number, in InvalidParameterArgument()
H A Dsort_simplifier.cc85 int64_t parameter_number = 0; in RemoveUnusedOperandFromSort() local
H A Dlocal_service.cc80 const XlaComputation& computation, int parameter_number) { in ParameterMetadata()
H A Dwhile_loop_all_reduce_code_motion.cc260 int parameter_number = in IsAllReduceMovable() local
H A Dhlo_instructions.h1392 int64_t parameter_number() const { return parameter_number_; } in parameter_number() function
H A Dhlo_dataflow_analysis.cc1268 for (int64_t parameter_number = 0; in Propagate() local
H A Dhlo_instruction.cc1008 int64_t parameter_number, const Shape& shape, const std::string& name) { in CreateParameter()
4635 int64_t HloInstruction::parameter_number() const { in parameter_number() function in xla::HloInstruction
H A Dhlo_verifier.cc759 const HloComputation* computation, int64_t parameter_number) { in CheckOperandAndParameter()
H A Dhlo_parser.cc1272 int64_t parameter_number; in CreateInstruction() local
H A Dhlo_instructions.cc2199 HloParameterInstruction::HloParameterInstruction(int64_t parameter_number, in HloParameterInstruction()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dclient_library_test_base.h606 NativeT value, int64_t parameter_number, const std::string& name, in CreateR0Parameter()
619 absl::Span<const NativeT> values, int64_t parameter_number, in CreateR1Parameter()
632 const Array2D<NativeT>& array_2d, int64_t parameter_number, in CreateR2Parameter()
645 const Array3D<NativeT>& array_3d, int64_t parameter_number, in CreateR3Parameter()
658 const Array4D<NativeT>& array_4d, int64_t parameter_number, in CreateR4Parameter()
671 const Array<NativeT>& array, int64_t parameter_number, in CreateParameter()
H A Dclient_library_test_base.cc612 int64_t parameter_number, const Literal& literal, const std::string& name, in CreateParameterAndTransferLiteral()
642 int64_t parameter_number, const Literal& literal, const std::string& name, in CreateParameterAndTransferLiteral()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dfusion_bitcast_lift.cc154 int64_t parameter_number = i->parameter_number(); in Run() local
/aosp_15_r20/external/sdv/vsomeip/third_party/boost/container/src/
Ddlmalloc.cpp61 BOOST_CONTAINER_DECL int dlmalloc_mallopt(int parameter_number, int parameter_value) in dlmalloc_mallopt()
/aosp_15_r20/external/clang/test/SemaCXX/
H A Daddr-of-overloaded-function.cpp182 void parameter_number() { in parameter_number() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dxla_builder.h448 XlaOp Parameter(int64_t parameter_number, const Shape& shape, in Parameter()
H A Dxla_builder.cc911 int64_t parameter_number, const Shape& shape, const std::string& name, in Parameter()
4034 XlaOp Parameter(XlaBuilder* builder, int64_t parameter_number, in Parameter()
4040 XlaOp Parameter(XlaBuilder* builder, int64_t parameter_number, in Parameter()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dspmd_partitioner_util.cc1072 auto match_bitcast_f32 = [](int64_t parameter_number) { in IsNanSafeGt()
1087 auto match_bitcast_bf16 = [](int64_t parameter_number) { in IsNanSafeGt()