Home
last modified time | relevance | path

Searched refs:parameter_num (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_parameter_binding.cc47 binding_proto.set_dynamic_param_num(dynamic_param.parameter_num); in ToProto()
52 binding_proto.set_target_param_num(dynamic_dimension.parameter_num); in ToProto()
95 dynamic_dimension.parameter_num, in ToString()
97 dynamic_dimension.dimension, dynamic_param.parameter_num, in ToString()
115 TF_RET_CHECK(dynamic_parameter.parameter_num >= 0 && in Verify()
116 dynamic_parameter.parameter_num < entry->num_parameters()); in Verify()
117 TF_RET_CHECK(dynamic_dimension.parameter_num < entry->num_parameters()); in Verify()
119 entry->parameter_instruction(dynamic_parameter.parameter_num)->shape(), in Verify()
122 entry->parameter_instruction(dynamic_dimension.parameter_num)->shape(), in Verify()
127 entry->parameter_instruction(dynamic_dimension.parameter_num) in Verify()
Ddynamic_parameter_binding.h46 int64 parameter_num; member
56 int64 parameter_num; member
65 return H::combine(std::move(h), m.parameter_num, m.parameter_index, in AbslHashValue()
71 return lhs.parameter_num == rhs.parameter_num &&
Ddynamic_parameter_binding_test.cc73 EXPECT_EQ(param->parameter_num, 0); in TEST_F()
112 EXPECT_EQ(param->parameter_num, 0); in TEST_F()
155 EXPECT_EQ(param->parameter_num, 0); in TEST_F()
165 EXPECT_EQ(param2->parameter_num, 0); in TEST_F()
Dpattern_matcher.h1604 explicit constexpr HloInstructionPatternParameterNumImpl(int64 parameter_num)
1605 : parameter_num_(parameter_num) {}
1985 constexpr auto WithParameterNum(int64 parameter_num) const -> decltype(
1986 this->AppendImpl(HloInstructionPatternParameterNumImpl(parameter_num))) {
1987 return AppendImpl(HloInstructionPatternParameterNumImpl(parameter_num));
2403 inline auto Parameter(int64 parameter_num) -> decltype(
2404 Op().WithOpcode(HloOpcode::kParameter).WithParameterNum(parameter_num)) {
2405 return Op().WithOpcode(HloOpcode::kParameter).WithParameterNum(parameter_num);
2408 inline auto Parameter(HloInstructionType** matched_inst, int64 parameter_num)
2411 .WithParameterNum(parameter_num)) {
[all …]
Ddynamic_dimension_inference.cc1258 if (dynamic_dimension.parameter_num != hlo->parameter_number()) { in HandleParameter()
1263 computation->parameter_instruction(dynamic_dimension.parameter_num); in HandleParameter()
1266 computation->parameter_instruction(dynamic_parameter.parameter_num); in HandleParameter()
Ddynamic_padder.cc1000 dynamic_dimension.parameter_num); in Run()
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client.py883 def ParameterWithShape(self, shape, name=None, parameter_num=None, argument
902 if parameter_num is None:
903 parameter_num = next(self._parameter_numbering)
907 return ops.Parameter(self._builder, parameter_num,
911 def ParameterFromNumpy(self, value, name=None, parameter_num=None): argument
924 shape_from_pyval(value), name=name, parameter_num=parameter_num)
Dxla_client_test.py412 p1 = c.ParameterFromNumpy(self.f32_4vector, parameter_num=1)
413 p0 = c.ParameterFromNumpy(self.f32_scalar_2, parameter_num=0)
425 p1 = c.ParameterFromNumpy(self.f64_4vector, parameter_num=1)
426 p0 = c.ParameterFromNumpy(self.f64_scalar_2, parameter_num=0)