Searched refs:input_tensor_ (Results 1 – 2 of 2) sorted by relevance
431 input_tensor_ = in InitTensor()434 init_f(&input_tensor_); in InitTensor()442 &input_tensor_) in InitTensor()456 inputs.push_back(TensorValue(&input_tensor_)); in DoGather()473 *col_params_, &input_tensor_, DEVICE_CPU, device_); in DoGather()481 parent_->instances_.size() * input_tensor_.shape().num_elements())}); in DoGather()493 kStepId, &input_tensor_, output_tensor_ptr); in DoGather()506 const Tensor& input_tensor() { return input_tensor_; } in input_tensor()513 Tensor input_tensor_; member in tensorflow::RingGathererTest::DeviceInstance
103 TfLiteIntArrayFree(input_tensor_.dims); in ~BaseLstmParam()152 TfLiteTensor input_tensor_; member in tflite::__anon39d466b60111::BaseLstmParam324 PackWeightToTensor(&input_tensor_, input_, input_size_); in GetInput()325 input_tensor_.data.int8 = input_.data(); in GetInput()326 return &input_tensor_; in GetInput()722 PackWeightToTensor(&input_tensor_, input_float_, input_size_); in GetFloatInput()723 input_tensor_.data.f = input_float_.data(); in GetFloatInput()724 return &input_tensor_; in GetFloatInput()