Searched refs:OpConverterParams (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 462 Status CreateBroadcastableScalarConstant(OpConverterParams* params, float value, in CreateBroadcastableScalarConstant()
974 OpConverterParams params( in ValidateNode()
985 OpConverterParams params( in ConvertConstToWeights()
1005 OpConverterParams params(this, node_def, inputs, &outputs, in ConvertNode()
1427 const OpConverterParams& params, in CheckInputsWeights()
1455 Status AllowDataTypes(const OpConverterParams& params, in AllowDataTypes()
1571 Status BinaryTensorOpWeight(OpConverterParams* params, in BinaryTensorOpWeight()
1763 Status ConvertConv2DHelper(OpConverterParams* params, int group, in ConvertConv2DHelper()
1935 Status BinaryTensorOpTensor(OpConverterParams* params, in BinaryTensorOpTensor()
2004 Status ConvertPlugin(OpConverterParams* params) { in ConvertPlugin()
[all …]
|
| D | convert_nodes.h | 347 struct OpConverterParams { struct
348 OpConverterParams(Converter* arg_converter, const NodeDef& arg_node_def, in OpConverterParams() argument
367 using OpConverter = std::function<Status(OpConverterParams*)>; argument
|
| D | convert_nodes_test.cc | 486 &should_fail](OpConverterParams* params) -> Status { in TEST_F()
581 auto op_converter = [&output_tensors](OpConverterParams* params) -> Status { in TEST_F()
649 auto op_converter = [&output_tensors](OpConverterParams* params) -> Status { in TEST_F()
|