Home
last modified time | relevance | path

Searched refs:inputTensor (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/lite/objc/tests/
DTFLInterpreterTests.m132 TFLTensor *inputTensor = [customInterpreter inputTensorAtIndex:0 error:&error];
133 XCTAssertNotNil(inputTensor);
135 XCTAssertTrue([inputTensor.name isEqualToString:@"input"]);
136 XCTAssertEqual(inputTensor.dataType, TFLTensorDataTypeFloat32);
137 NSArray<NSNumber *> *inputTensorShape = [inputTensor shapeWithError:&error];
147 XCTAssertTrue([inputTensor copyData:inputData error:&error]);
220 TFLTensor *inputTensor = [customInterpreter inputTensorAtIndex:0 error:&error];
221 XCTAssertNotNil(inputTensor);
223 XCTAssertTrue([inputTensor.name isEqualToString:@"input"]);
224 XCTAssertEqual(inputTensor.dataType, TFLTensorDataTypeUInt8);
[all …]
DTFLCoreMLDelegateTests.m83 TFLTensor* inputTensor = [interpreter inputTensorAtIndex:0 error:&error];
84 XCTAssertNotNil(inputTensor);
85 XCTAssertTrue([inputTensor copyData:inputData error:&error]);
DTFLMetalDelegateTests.m91 TFLTensor* inputTensor = [interpreter inputTensorAtIndex:input_idx error:&error];
92 XCTAssertNotNil(inputTensor);
93 XCTAssertTrue([inputTensor copyData:inputData error:&error]);
/external/tensorflow/tensorflow/lite/swift/Tests/
DTensorTests.swift27 let inputTensor = Tensor( in testInit() variable
34 XCTAssertEqual(inputTensor.name, name) in testInit()
35 XCTAssertEqual(inputTensor.dataType, dataType) in testInit()
36 XCTAssertEqual(inputTensor.shape, shape) in testInit()
37 XCTAssertEqual(inputTensor.data, data) in testInit()
38 XCTAssertEqual(inputTensor.quantizationParameters, quantizationParameters) in testInit()
DInterpreterTests.swift74 let inputTensor = try interpreter.input(at: AddModel.validIndex) in testInputTensorAtIndex() variable
75 XCTAssertEqual(inputTensor, AddModel.inputTensor) in testInputTensorAtIndex()
81 let inputTensor = try interpreter.input(at: AddQuantizedModel.inputOutputIndex) in testInputTensorAtIndex_QuantizedModel() variable
82 XCTAssertEqual(inputTensor, AddQuantizedModel.inputTensor) in testInputTensorAtIndex_QuantizedModel()
162 let inputTensor = try interpreter.copy(AddModel.inputData, toInputAt: AddModel.validIndex) in testCopyDataToInputTensorAtIndex() variable
163 XCTAssertEqual(inputTensor.data, AddModel.inputData) in testCopyDataToInputTensorAtIndex()
286 static let inputTensor = Tensor( variable
316 static let inputTensor = Tensor( variable
/external/tensorflow/tensorflow/lite/objc/apps/TestApp/TestApp/
DViewController.m198 TFLTensor *inputTensor = [weakSelf.interpreter inputTensorAtIndex:0 error:&error];
199 if (inputTensor == nil || error != nil) {
210 if (![inputTensor copyData:inputData error:&error]) {
268 TFLTensor *inputTensor = [weakSelf.interpreter inputTensorAtIndex:0 error:&error];
269 if (inputTensor == nil || error != nil) {
280 if (![inputTensor copyData:inputData error:&error]) {
355 TFLTensor *inputTensor = [weakSelf.interpreter inputTensorAtIndex:i error:&error];
356 if (inputTensor == nil || error != nil) {
370 if (![inputTensor copyData:inputData error:&error]) {
/external/tensorflow/tensorflow/lite/java/src/main/java/org/tensorflow/lite/
DNativeInterpreterWrapper.java360 Tensor inputTensor = inputTensors[index]; in getInputTensor() local
361 if (inputTensor == null) { in getInputTensor()
362 inputTensor = in getInputTensor()
366 return inputTensor; in getInputTensor()
/external/tensorflow/tensorflow/lite/java/src/test/java/org/tensorflow/lite/
DInterpreterTest.java746 Tensor inputTensor = interpreter.getInputTensorFromSignature("x", "mul_add"); in testModelWithSignatureDef() local
748 assertThat(inputTensor.numElements()).isEqualTo(1); in testModelWithSignatureDef()
752 inputTensor = interpreter.getInputTensorFromSignature(null, "mul_add"); in testModelWithSignatureDef()
759 inputTensor = interpreter.getInputTensorFromSignature("xx", "mul_add"); in testModelWithSignatureDef()
/external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
DFusion.cpp493 Value inputTensor = consumer.getInput(consumerIdx); in fuseProducerOfTensor() local
496 getProducerOfTensor(inputTensor, producerOp, producerIdx); in fuseProducerOfTensor()
499 auto subTensor = inputTensor.getDefiningOp<SubTensorOp>(); in fuseProducerOfTensor()