Searched refs:tensor_item (Results 1 – 4 of 4) sorted by relevance
924 void AddTensorProtoInfo(TensorProto *tensor_item, const TensorProto &tensor) { in AddTensorProtoInfo() argument925 tensor_item->set_node_name(tensor.node_name()); in AddTensorProtoInfo()926 tensor_item->set_slot(tensor.slot()); in AddTensorProtoInfo()927 tensor_item->set_iter(tensor.iter()); in AddTensorProtoInfo()928 tensor_item->set_truncate(tensor.truncate()); in AddTensorProtoInfo()929 tensor_item->clear_tensor_content(); in AddTensorProtoInfo()930 tensor_item->clear_data_type(); in AddTensorProtoInfo()931 tensor_item->clear_dims(); in AddTensorProtoInfo()1003 TensorProto tensor_item; in LoadTensors() local1004 tensor_item.set_finished(true); in LoadTensors()[all …]
721 … std::back_inserter(tensor_shape), [](Tensor *tensor_item) { return tensor_item->shape(); }); in SetInputShape() argument
142 def tensor_item(data, *args): function181 tensor_operator_registry.register("item", tensor_item)
1418 return compile_utils.tensor_item(data, *args)