Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dfused_ir_emitter.h106 llvm_ir::IrArray& GetIrArrayForFusedParameter(int64 parameter_number) { in GetIrArrayForFusedParameter()
113 llvm::Value* GetBasePointerForFusedParameter(int64 parameter_number) { in GetBasePointerForFusedParameter()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dclient_library_test_base.h566 NativeT value, int64 parameter_number, const string& name, in CreateR0Parameter()
580 absl::Span<const NativeT> values, int64 parameter_number, in CreateR1Parameter()
594 const Array2D<NativeT>& array_2d, int64 parameter_number, in CreateR2Parameter()
608 const Array3D<NativeT>& array_3d, int64 parameter_number, in CreateR3Parameter()
622 const Array4D<NativeT>& array_4d, int64 parameter_number, in CreateR4Parameter()
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/tensorflow/tensorflow/compiler/xla/service/
Dsource_map_util.h48 Status InvalidParameterArgument(Executable* executable, int parameter_number, in InvalidParameterArgument()
Dhlo_input_output_alias_config.h53 int64 parameter_number; member
Dhlo_matchers.h46 explicit HloParameterMatcher(int64 parameter_number) in HloParameterMatcher()
288 int64 parameter_number) { in Parameter()
Dsort_simplifier.cc85 int64 parameter_number = 0; in RemoveUnusedOperandFromSort() local
Dhlo.proto69 int64 parameter_number = 9; field
309 int64 parameter_number = 2; field
416 int64 parameter_number = 6; field
Dlocal_service.cc80 const XlaComputation& computation, int parameter_number) { in ParameterMetadata()
Dbuffer_assignment.h127 int64 parameter_number() const { in parameter_number() function
283 void set_entry_computation_parameter(int64 parameter_number, in set_entry_computation_parameter()
Dhlo_instructions.h932 int64 parameter_number() const { return parameter_number_; } in parameter_number() function
Dhlo_verifier.cc321 const HloComputation* computation, int64 parameter_number) { in CheckOperandAndParameter()
Dhlo_instruction.cc708 int64 parameter_number, const Shape& shape, const string& name) { in CreateParameter()
3700 int64 HloInstruction::parameter_number() const { in parameter_number() function in xla::HloInstruction
Dhlo_parser.cc716 int64 parameter_number; in ParseInstructionRhs() local
Dhlo_instructions.cc1759 HloParameterInstruction::HloParameterInstruction(int64 parameter_number, in HloParameterInstruction()
/external/clang/test/SemaCXX/
Daddr-of-overloaded-function.cpp182 void parameter_number() { in parameter_number() function
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h370 XlaOp Parameter(int64 parameter_number, const Shape& shape, in Parameter()
Dxla_builder.cc737 int64 parameter_number, const Shape& shape, const string& name, in Parameter()
2875 XlaOp Parameter(XlaBuilder* builder, int64 parameter_number, const Shape& shape, in Parameter()
2881 XlaOp Parameter(XlaBuilder* builder, int64 parameter_number, const Shape& shape, in Parameter()