Searched refs:tensors_mask_ (Results 1 – 4 of 4) sorted by relevance
39 tensors_mask_(std::move(tensors_mask)), in RunOpContext()46 const std::vector<int64_t> &tensor_mask() const { return tensors_mask_; } in tensor_mask()54 std::vector<int64_t> tensors_mask_; variable
157 session_->RunOpImpl(graph_info_, op_run_info_, input_tensors_, &outputs_, tensors_mask_); in Run()454 task->tensors_mask_ = tensors_mask; in RunOp()
121 std::vector<int64_t> tensors_mask_; variable
68 tensors_mask_(tensors_mask), in GraphCompilerInfo()81 std::vector<std::vector<int64_t> *> tensors_mask_; member