Searched refs:start_array (Results 1 – 8 of 8) sorted by relevance
24 void FillRangeOutput(const Array& start_array, const Array& limit_array, in FillRangeOutput() argument27 T start = start_array.GetBuffer<A>().data[0]; in FillRangeOutput()55 const auto& start_array = model->GetArray(op->inputs[0]); in Run() local56 if (!start_array.has_shape()) { in Run()85 CHECK_EQ(RequiredBufferSizeForShape(start_array.shape()), 1) in Run()92 CHECK(start_array.data_type == ArrayDataType::kInt32 || in Run()93 start_array.data_type == ArrayDataType::kFloat) in Run()95 CHECK(limit_array.data_type == start_array.data_type) in Run()97 CHECK(delta_array.data_type == start_array.data_type) in Run()100 if (start_array.data_type == ArrayDataType::kInt32) { in Run()[all …]
62 auto& start_array = model->GetArray(op->inputs[1]); in Run() local63 if (!start_array.has_shape()) return ::tensorflow::Status::OK(); in Run()64 if (toco::RequiredBufferSizeForShape(start_array.shape()) > 4) { in Run()83 int start_indices_size = start_array.shape().dims(0); in Run()112 PadAttributeArray(&start_array, begin_pad_values, op->begin_mask); in Run()118 op->start_indices = start_array.GetBuffer<ArrayDataType::kInt32>().data; in Run()
698 const auto& start_array = model->GetArray(op->inputs[0]); in ProcessRangeOperator() local699 if (!start_array.has_shape()) { in ProcessRangeOperator()723 const ArrayDataType& start_dtype = start_array.data_type; in ProcessRangeOperator()733 CHECK_EQ(RequiredBufferSizeForShape(start_array.shape()), 1) in ProcessRangeOperator()743 start_array.GetBuffer<ArrayDataType::kInt32>().data[0]) / in ProcessRangeOperator()747 start_array.GetBuffer<ArrayDataType::kFloat>().data[0]) / in ProcessRangeOperator()
267 def start_array(self: PlistTarget) -> None: function310 "array": start_array,
2166 static bool start_array(upb_json_parser *p) { in start_array() function2404 start_array(p); in start_fieldmask_object()3142 { CHECK_RETURN_TOP(start_array(parser)); } in parse()
2164 static bool start_array(upb_json_parser *p) {2402 start_array(p);2754 >{ CHECK_RETURN_TOP(start_array(parser)); }
10948 static bool start_array(upb_json_parser *p) { in start_array() function11186 start_array(p); in start_fieldmask_object()11924 { CHECK_RETURN_TOP(start_array(parser)); } in parse()
14798 static bool start_array(upb_json_parser *p) { in start_array() function15046 start_array(p); in start_fieldmask_object()15788 { CHECK_RETURN_TOP(start_array(parser)); } in parse()