Home
last modified time | relevance | path

Searched refs:OpConverterParams (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.h358 struct OpConverterParams { struct
360 OpConverterParams(const NodeDef& node_def,
368 OpConverterParams(Converter* converter, const NodeDef& node_def,
384 using OpConverter = std::function<Status(OpConverterParams*)>; argument
575 OpConverterParams* params, nvinfer1::ITensor** output,
581 int axis, OpConverterParams* params,
590 OpConverterParams* params, nvinfer1::ITensor** output);
Dconvert_nodes.cc492 OpConverterParams* params, T value, nvinfer1::ITensor** tensor, in CreateScalarConstant()
506 Status CreateBroadcastableScalarConstant(OpConverterParams* params, float value, in CreateBroadcastableScalarConstant()
1101 OpConverterParams::OpConverterParams( in OpConverterParams() function in tensorflow::tensorrt::convert::OpConverterParams
1115 OpConverterParams::OpConverterParams( in OpConverterParams() function in tensorflow::tensorrt::convert::OpConverterParams
1222 OpConverterParams params(node->def(), inputs, /*arg_outputs=*/nullptr, in IsTensorRTCandidate()
1233 OpConverterParams params(const_node_def, inputs, &outputs, &weight_store_, in ConvertConstToWeights()
1290 OpConverterParams params(this, node_def, inputs, &outputs, &weight_store_); in ConvertNode()
2051 const OpConverterParams& params, in CheckInputsWeights()
2085 const OpConverterParams& params, in CheckInputsWeights()
2109 Status GetInputTfType(const OpConverterParams& params, DataType* tf_type, in GetInputTfType()
[all …]
Dconvert_nodes_test.cc635 &should_fail](OpConverterParams* params) -> Status { in TEST_F()
795 auto op_converter = [&output_tensors](OpConverterParams* params) -> Status { in TEST_F()
867 auto op_converter = [&output_tensors](OpConverterParams* params) -> Status { in TEST_F()