Searched refs:op_reg_data_ (Results 1 – 3 of 3) sorted by relevance
166 OpDef* op_def() { return &op_reg_data_.op_def; } in op_def()168 OpRegistrationData op_reg_data_; variable
629 if (op_reg_data_.shape_inference_fn != nullptr) { in SetShapeFn()633 op_reg_data_.shape_inference_fn = OpShapeInferenceFn(fn); in SetShapeFn()645 *op_reg_data = op_reg_data_; in Finalize()
203 return op_reg_data_; in op_reg_data()240 tensorflow::OpRegistry::Global()->LookUp(nodedef_.op(), &op_reg_data_)); in InitializeNodeDef()241 AddDefaultsToNodeDef(op_reg_data_->op_def, &nodedef_); in InitializeNodeDef()326 const tensorflow::OpRegistrationData* op_reg_data_; member in tflite::flex::OpNode