Home
last modified time | relevance | path

Searched refs:OpShapeInferenceFn (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dop_def_builder.h37 OpShapeInferenceFn; typedef
43 OpRegistrationData(const OpDef& def, const OpShapeInferenceFn& fn,
48 OpShapeInferenceFn shape_inference_fn;
143 OpDefBuilder& SetShapeFn(OpShapeInferenceFn fn);
Dshape_inference_testutil_test.cc35 static OpShapeInferenceFn* global_fn_ptr = nullptr;
51 const string& expected_outs, OpShapeInferenceFn fn) { in RunInferShapes()
Dop.h262 OpDefBuilderWrapper& SetShapeFn(OpShapeInferenceFn fn) { in SetShapeFn()
Dop_def_builder.cc628 OpDefBuilder& OpDefBuilder::SetShapeFn(OpShapeInferenceFn fn) { in SetShapeFn()
633 op_reg_data_.shape_inference_fn = OpShapeInferenceFn(fn); in SetShapeFn()
Dop_def_builder_test.cc44 OpShapeInferenceFn* shape_fn_out = nullptr) { in ExpectSuccess()
/external/tensorflow/tensorflow/c/
Dops.cc28 using ::tensorflow::OpShapeInferenceFn;