Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmodel_builder.cc77 const ParamsT** tf_options) { in RetrieveBuiltinData() argument
78 *tf_options = static_cast<const ParamsT*>(tflite_node->builtin_data); in RetrieveBuiltinData()
79 if (!*tf_options) { in RetrieveBuiltinData()
183 const TfLiteAddParams* tf_options; in IsSupported() local
184 return RetrieveBuiltinData(tflite_node, &tf_options); in IsSupported()
200 const TfLiteAddParams* tf_options; in Parse() local
201 RETURN_IF_ERROR(RetrieveBuiltinData(tflite_node, &tf_options)); in Parse()
202 return MaybeFuseActivation(tf_options->activation, graph, node); in Parse()
239 const TfLiteConcatenationParams* tf_options; in IsSupported() local
240 RETURN_IF_ERROR(RetrieveBuiltinData(tflite_node, &tf_options)); in IsSupported()
[all …]
Doperation_parser.cc77 absl::Status ParsePoolingAttributes(const TfLitePoolParams* tf_options, in ParsePoolingAttributes() argument
80 attr->kernel = ToHW(tf_options->filter_height, tf_options->filter_width); in ParsePoolingAttributes()
81 attr->strides = ToHW(tf_options->stride_height, tf_options->stride_width); in ParsePoolingAttributes()
82 UpdatePadding(tf_options->padding, input_shape, attr); in ParsePoolingAttributes()
Doperation_parser.h61 absl::Status ParsePoolingAttributes(const TfLitePoolParams* tf_options,
/external/tensorflow/tensorflow/compiler/mlir/tfjs/
Dtf_tfjs_passes.cc31 mlir::TF::StandardPipelineOptions tf_options; in AddTFToTFJSConversionPasses() local
32 tf_options.enable_inliner = true; in AddTFToTFJSConversionPasses()
33 mlir::TF::CreateTFStandardPipeline(*pm, tf_options); in AddTFToTFJSConversionPasses()
/external/tensorflow/tensorflow/compiler/mlir/python/
Dmlir.cc221 mlir::TF::StandardPipelineOptions tf_options; in ExperimentalConvertSavedModelV1ToMlir() local
222 mlir::TF::CreateTFStandardPipeline(pm, tf_options); in ExperimentalConvertSavedModelV1ToMlir()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dcompile_mlir_util.cc603 mlir::TF::StandardPipelineOptions tf_options; in CompileGraphSetup() local
604 mlir::TF::CreateTFStandardPipeline(pm, tf_options); in CompileGraphSetup()