Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
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()
Dhlo_input_output_alias_config.h47 : parameter_number(parameter_number), in parameter_number() function
51 int64_t parameter_number; member
Dsource_map_util.h47 Status InvalidParameterArgument(Executable* executable, int parameter_number, in InvalidParameterArgument()
Dhlo_matchers.h49 explicit HloParameterMatcher(int64_t parameter_number) in HloParameterMatcher()
342 int64_t parameter_number) { in Parameter()
Dsort_simplifier.cc85 int64_t parameter_number = 0; in RemoveUnusedOperandFromSort() local
Dlocal_service.cc80 const XlaComputation& computation, int parameter_number) { in ParameterMetadata()
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()
Dhlo.proto124 int64 parameter_number = 9; field
411 int64 parameter_number = 2; field
563 int64 parameter_number = 6; field
Dwhile_loop_all_reduce_code_motion.cc260 int parameter_number = in IsAllReduceMovable() local
Dhlo_instructions.h1392 int64_t parameter_number() const { return parameter_number_; } in parameter_number() function
Dhlo_dataflow_analysis.cc1268 for (int64_t parameter_number = 0; in Propagate() local
Dhlo_verifier.cc759 const HloComputation* computation, int64_t parameter_number) { in CheckOperandAndParameter()
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
Dhlo_parser.cc1272 int64_t parameter_number; in CreateInstruction() local
Dhlo_instructions.cc2199 HloParameterInstruction::HloParameterInstruction(int64_t parameter_number, in HloParameterInstruction()
/external/tensorflow/tensorflow/compiler/xla/tests/
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()
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()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dfusion_bitcast_lift.cc154 int64_t parameter_number = i->parameter_number(); in Run() local
/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()
/external/clang/test/SemaCXX/
Daddr-of-overloaded-function.cpp182 void parameter_number() { in parameter_number() function
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h448 XlaOp Parameter(int64_t parameter_number, const Shape& shape, in Parameter()
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()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
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()