Searched refs:tf_options (Results 1 – 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
D | model_builder.cc | 77 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 …]
|
D | operation_parser.cc | 77 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()
|
D | operation_parser.h | 61 absl::Status ParsePoolingAttributes(const TfLitePoolParams* tf_options,
|
/external/tensorflow/tensorflow/compiler/mlir/tfjs/ |
D | tf_tfjs_passes.cc | 31 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/ |
D | mlir.cc | 221 mlir::TF::StandardPipelineOptions tf_options; in ExperimentalConvertSavedModelV1ToMlir() local 222 mlir::TF::CreateTFStandardPipeline(pm, tf_options); in ExperimentalConvertSavedModelV1ToMlir()
|
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
D | compile_mlir_util.cc | 603 mlir::TF::StandardPipelineOptions tf_options; in CompileGraphSetup() local 604 mlir::TF::CreateTFStandardPipeline(pm, tf_options); in CompileGraphSetup()
|