Searched refs:delta_array (Results 1 – 2 of 2) sorted by relevance
24 const Array& delta_array, Array* output_array) { in FillRangeOutput() argument28 T delta = delta_array.GetBuffer<A>().data[0]; in FillRangeOutput()64 const auto& delta_array = model->GetArray(op->inputs[2]); in Run() local65 if (!delta_array.has_shape()) { in Run()88 CHECK_EQ(RequiredBufferSizeForShape(delta_array.shape()), 1) in Run()96 CHECK(delta_array.data_type == start_array.data_type) in Run()101 delta_array, &output_array); in Run()104 delta_array, &output_array); in Run()
706 const auto& delta_array = model->GetArray(op->inputs[2]); in ProcessRangeOperator() local707 if (!delta_array.has_shape()) { in ProcessRangeOperator()726 CHECK(delta_array.data_type == ArrayDataType::kInt32) in ProcessRangeOperator()732 CHECK_EQ(RequiredBufferSizeForShape(delta_array.shape()), 1) in ProcessRangeOperator()736 delta_array.GetBuffer<ArrayDataType::kInt32>().data[0]); in ProcessRangeOperator()