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.h104 llvm_ir::IrArray& GetIrArrayForFusedParameter(int64 parameter_number) { in GetIrArrayForFusedParameter()
111 llvm::Value* GetBasePointerForFusedParameter(int64 parameter_number) { in GetBasePointerForFusedParameter()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dclient_library_test_base.h550 NativeT value, int64 parameter_number, const string& name, in CreateR0Parameter()
564 absl::Span<const NativeT> values, int64 parameter_number, in CreateR1Parameter()
578 const Array2D<NativeT>& array_2d, int64 parameter_number, in CreateR2Parameter()
592 const Array3D<NativeT>& array_3d, int64 parameter_number, in CreateR3Parameter()
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()
260 int64 parameter_number) { in Parameter()
Dhlo.proto63 int64 parameter_number = 9; field
277 int64 parameter_number = 2; field
384 int64 parameter_number = 6; field
Dsort_simplifier.cc85 int64 parameter_number = 0; in RemoveUnusedOperandFromSort() local
Dlocal_service.cc80 const XlaComputation& computation, int parameter_number) { in ParameterMetadata()
Dbuffer_assignment.h124 int64 parameter_number() const { in parameter_number() function
280 void set_entry_computation_parameter(int64 parameter_number, in set_entry_computation_parameter()
Dhlo_instructions.h866 int64 parameter_number() const { return parameter_number_; } in parameter_number() function
Dhlo_verifier.cc234 const HloComputation* computation, int64 parameter_number) { in CheckOperandAndParameter()
Dhlo_instruction.cc638 int64 parameter_number, const Shape& shape, const string& name) { in CreateParameter()
3367 int64 HloInstruction::parameter_number() const { in parameter_number() function in xla::HloInstruction
Dhlo_parser.cc687 int64 parameter_number; in ParseInstructionRhs() local
Dhlo_instructions.cc1638 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/python/
Dlocal_computation_builder.cc427 LocalOp ComputationBuilder::Parameter(int64 parameter_number, in Parameter()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc632 XlaOp XlaBuilder::Parameter(int64 parameter_number, const Shape& shape, in Parameter()
2828 XlaOp Parameter(XlaBuilder* builder, int64 parameter_number, const Shape& shape, in Parameter()