Searched refs:parameter_num (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_parameter_binding.cc | 47 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()
|
| D | dynamic_parameter_binding.h | 46 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 &&
|
| D | dynamic_parameter_binding_test.cc | 74 EXPECT_EQ(param->parameter_num, 0); in TEST_F()
113 EXPECT_EQ(param->parameter_num, 0); in TEST_F()
156 EXPECT_EQ(param->parameter_num, 0); in TEST_F()
166 EXPECT_EQ(param2->parameter_num, 0); in TEST_F()
|
| D | pattern_matcher.h | 1588 explicit constexpr HloInstructionPatternParameterNumImpl(int64 parameter_num)
1589 : parameter_num_(parameter_num) {}
1973 constexpr auto WithParameterNum(int64 parameter_num) const -> decltype(
1974 this->AppendImpl(HloInstructionPatternParameterNumImpl(parameter_num))) {
1975 return AppendImpl(HloInstructionPatternParameterNumImpl(parameter_num));
2389 inline auto Parameter(int64 parameter_num) -> decltype(
2390 Op().WithOpcode(HloOpcode::kParameter).WithParameterNum(parameter_num)) {
2391 return Op().WithOpcode(HloOpcode::kParameter).WithParameterNum(parameter_num);
2394 inline auto Parameter(HloInstructionType** matched_inst, int64 parameter_num)
2397 .WithParameterNum(parameter_num)) {
[all …]
|
| D | dynamic_dimension_inference.cc | 532 if (dynamic_dimension.parameter_num != hlo->parameter_number()) { in HandleParameter()
537 computation->parameter_instruction(dynamic_dimension.parameter_num); in HandleParameter()
540 computation->parameter_instruction(dynamic_parameter.parameter_num); in HandleParameter()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 1029 def ParameterWithShape(self, shape, name=None, parameter_num=None): argument
1045 if parameter_num is None:
1046 parameter_num = next(self._parameter_numbering)
1048 return self._client.Parameter(parameter_num, shape, name.encode('utf8'))
1050 def ParameterFromNumpy(self, value, name=None, parameter_num=None): argument
1063 Shape.from_pyval(value), name=name, parameter_num=parameter_num)
|
| D | xla_client_test.py | 388 p1 = c.ParameterFromNumpy(self.f32_4vector, parameter_num=1)
389 p0 = c.ParameterFromNumpy(self.f32_scalar_2, parameter_num=0)
401 p1 = c.ParameterFromNumpy(self.f64_4vector, parameter_num=1)
402 p0 = c.ParameterFromNumpy(self.f64_scalar_2, parameter_num=0)
|