Searched refs:DataVec (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| D | trt_engine_utils.h | 122 using DataVec = std::vector<InputOutputData>; variable
139 const DataVec* input_vec = nullptr);
159 DataVec* outputs = nullptr);
|
| D | trt_engine_utils.cc | 139 const DataVec* input_vec) { in SetTrtEngineInputs()
221 OpKernelContext* ctx, DataVec* outputs) { in SetTrtEngineOutputs()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 1322 Status GetShapeFromDataVec(DataVec input_data, in GetShapeFromDataVec()
1449 void CheckDataTypeMatches(const DataVec& datas) { in CheckDataTypeMatches()
1463 Status BuildAndRun(const DataVec& input_data, DataVec* output_data, in BuildAndRun()
1876 DataVec output_data;
1948 DataVec input_data_;
2411 const DataVec input_data{{"input", AsTensor<float>(input_vec)}}; in TEST_F()
2412 DataVec output_data{ in TEST_F()
2516 const DataVec input_data{{"input", test->AsTensor<float>({0, 1})}}; in TestMatMulHelper()
2517 DataVec output_data{{"my_matmul", test->ConstructTensor<float>(2)}}; in TestMatMulHelper()
2543 const DataVec input_data{{"input", test->AsTensor<float>({0, 1})}}; in TestMatMulHelper()
[all …]
|
| /external/swiftshader/third_party/subzero/src/ |
| D | IceTargetLoweringMIPS32.cpp | 5593 auto *DataVec = llvm::dyn_cast<VariableVecOn32>(Value); in lowerStore() local
5594 for (SizeT i = 0; i < DataVec->ContainersPerVector; ++i) { in lowerStore()
5595 auto *DCont = legalizeToReg(DataVec->getContainers()[i]); in lowerStore()
|